- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for appendDialectRegistry (0.36 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/test_utils.cc
absl::StatusOr<OwningOpRef<ModuleOp>> GetMlirModuleFromString( absl::string_view module_string, MLIRContext* context) { DialectRegistry mlir_registry; RegisterCommonToolingDialects(mlir_registry); context->appendDialectRegistry(mlir_registry); OwningOpRef<ModuleOp> mlir_module; auto status = tensorflow::DeserializeMlirModule(module_string, context, &mlir_module); if (!status.ok()) { return status; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/logging_hooks_test.cc
"tensorflow/compiler/mlir/tf2xla/internal/testdata/"); } class LoggingHooksTest : public ::testing::Test { public: LoggingHooksTest() { mlir::RegisterCommonToolingDialects(registry_); context_.appendDialectRegistry(registry_); context_.loadAllAvailableDialects(); env_ = Env::Default(); test_group_name_ = "TestGroup"; test_dir_ = testing::TmpDir(); setenv("TF_DUMP_GRAPH_PREFIX", test_dir_.c_str(), 1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/cluster_tf_test.cc
} class SessionClusterTensorflowDialectTest : public ::testing::Test { public: SessionClusterTensorflowDialectTest() { mlir::RegisterCommonToolingDialects(registry_); context_.appendDialectRegistry(registry_); context_.loadAllAvailableDialects(); } absl::Status CreateMlirModule(std::string mlir_module_filename) { std::string mlir_module_path = TestDataPath() + mlir_module_filename;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:44:37 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/tf_dialect_to_executor_test.cc
} class TensorflowDialectToExecutorTest : public ::testing::Test { public: TensorflowDialectToExecutorTest() { mlir::RegisterCommonToolingDialects(registry_); context_.appendDialectRegistry(registry_); context_.loadAllAvailableDialects(); } absl::Status CreateMlirModule(std::string mlir_module_filename) { std::string mlir_module_path = TestDataPath() + mlir_module_filename;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:08:57 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf_test.cc
absl::StatusOr<OwningOpRef<ModuleOp>> GetMlirModuleFromString( absl::string_view module_string, MLIRContext* context) { mlir::DialectRegistry mlir_registry; RegisterAllTensorFlowDialects(mlir_registry); context->appendDialectRegistry(mlir_registry); OwningOpRef<ModuleOp> mlir_module; auto status = tensorflow::DeserializeMlirModule(module_string, context, &mlir_module); if (!status.ok()) { return status; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_dialect_to_executor_test.cc
} return count; } class TensorflowDialectToExecutorTest : public ::testing::Test { public: TensorflowDialectToExecutorTest() { mlir::RegisterCommonToolingDialects(registry_); context_.appendDialectRegistry(registry_); context_.loadAllAvailableDialects(); } absl::Status CreateMlirModule(std::string mlir_module_filename) { std::string mlir_module_path = TestDataPath() + mlir_module_filename;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 23:22:50 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/utils/bfloat16_type_test.cc
namespace { std::unique_ptr<MLIRContext> CreateContext() { auto context = std::make_unique<MLIRContext>(); DialectRegistry mlir_registry; RegisterCommonToolingDialects(mlir_registry); context->appendDialectRegistry(mlir_registry); return context; } TEST(IsLargeFloatTypeTest, scalars) { auto context = CreateContext(); EXPECT_FALSE(IsLargeFloatType(Float8E4M3FNType::get(context.get())));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 19 23:51:52 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types_test.cc
class LegalizeTfTypesTest : public Test { protected: void CreateModule(const char* module_string) { DialectRegistry mlir_registry; RegisterCommonToolingDialects(mlir_registry); context_.appendDialectRegistry(mlir_registry); TF_ASSERT_OK( tensorflow::DeserializeMlirModule(module_string, &context_, &module_)); pm_ = std::make_unique<mlir::PassManager>(&context_); pm_->addNestedPass<mlir::func::FuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 09:05:02 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/utils/utils.cc
registry.insert<mlir::TFL::TensorFlowLiteDialect, mlir::arith::ArithDialect, mlir::func::FuncDialect>(); mlir::func::registerAllExtensions(registry); context->appendDialectRegistry(registry); source_mgr->AddNewSourceBuffer(std::move(buffer), llvm::SMLoc()); return mlir::OwningOpRef<mlir::ModuleOp>( mlir::parseSourceFile<mlir::ModuleOp>(*source_mgr, context)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 03 03:47:03 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util_test.cc
absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> GetMlirModuleFromString( llvm::StringRef string, mlir::MLIRContext* context) { mlir::DialectRegistry mlir_registry; RegisterAllTensorFlowDialects(mlir_registry); context->appendDialectRegistry(mlir_registry); mlir::OwningOpRef<mlir::ModuleOp> mlir_module; auto status = tensorflow::DeserializeMlirModule(string, context, &mlir_module); if (!status.ok()) { return status; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 4.3K bytes - Viewed (0)