- Sort Score
- Result 10 results
- Languages All
Results 11 - 17 of 17 for loadAllAvailableDialects (0.47 sec)
-
tensorflow/compiler/mlir/tensorflow_to_stablehlo/tf_to_stablehlo_translate.cc
"TF Saved Model to Stablehlo converter\n"); mlir::DialectRegistry registry; RegisterAllTensorFlowDialects(registry); mlir::MLIRContext context(registry); context.loadAllAvailableDialects(); bool is_input_mlir_module = absl::EndsWith(input_path, ".mlir"); std::vector<std::string> exported_model_signatures_in_vector = absl::StrSplit(exported_model_signatures, ',');
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/cluster_tf_test.cc
public: FunctionClusterTensorflowDialectTest() { 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; mlir_module_ =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:44:37 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/tac_module.cc
if (!module_status.ok()) { return module_status.status(); } auto module = module_status->get(); auto* context = module->getContext(); context->appendDialectRegistry(registry_); context->loadAllAvailableDialects(); // Run TAC passes. auto status = RunTacPasses(&module, options_.debug_mode); if (!status.ok()) { return status; } return exporter_->Export(module); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/lower_cluster_to_runtime_ops_test.cc
class LowerClusterToRuntimeOpsTest : public ::testing::Test { public: LowerClusterToRuntimeOpsTest() { 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(), /*overwrite=*/1); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:44:37 UTC 2024 - 6.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/debug/debug_test.cc
registry.insert<mlir::func::FuncDialect>(); registry.insert<mlir::TFL::TensorFlowLiteDialect>(); return registry; }()) { context_.loadAllAvailableDialects(); mlir::OpBuilder builder(&context_); module_ = builder.create<mlir::ModuleOp>(builder.getUnknownLoc()); builder.setInsertionPointToStart(module_->getBody());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 11:15:16 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/debugging/mlir_dump_test.cc
dialects.insert<mlir::BuiltinDialect, mlir::func::FuncDialect, mlir::stablehlo::StablehloDialect>(); ctx_ = std::make_unique<mlir::MLIRContext>(dialects); ctx_->loadAllAvailableDialects(); } void TearDown() override { // Delete files in the test directory. std::vector<std::string> files; TF_ASSERT_OK(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 03:17:14 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
mlir::TFR::TFRDialect>(); // clang-format on mlir::func::registerAllExtensions(registry); mlir_ctx->appendDialectRegistry(registry); mlir_ctx->loadAllAvailableDialects(); // Load the TFR functions in a mlir::ModuleOp auto memory_buffer = llvm::MemoryBuffer::getMemBuffer( llvm::StringRef(tfr_raw_text.data(), tfr_raw_text.size())); llvm::SourceMgr source_mgr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0)