- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 21 for loadAllAvailableDialects (0.42 sec)
-
tensorflow/compiler/mlir/python/mlir_wrapper/mlir_wrapper.cc
m.def("preloadTensorFlowDialects", [](mlir::MLIRContext &context) { mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialects(registry); context.appendDialectRegistry(registry); context.loadAllAvailableDialects(); }); m.def("verify", [](std::string input) { llvm::SourceMgr SM = llvm::SourceMgr(); SM.AddNewSourceBuffer(llvm::MemoryBuffer::getMemBuffer(input), llvm::SMLoc());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 08:41:42 UTC 2023 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config_test.cc
public: absl::Status CreateMlirModule(std::string module_string = kMlirModuleStr) { TF_ASSIGN_OR_RETURN( module_, test::GetMlirModuleFromString(module_string, &context_)); context_.loadAllAvailableDialects(); return absl::OkStatus(); } absl::StatusOr<FuncOp> GetMain() { func::FuncOp main = module_->lookupSymbol<mlir::func::FuncOp>("main"); if (!main) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 30 03:31:01 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow_to_stablehlo/python/pywrap_tensorflow_to_stablehlo_lib.cc
absl::string_view input_arg_shapes_str) { mlir::DialectRegistry registry; RegisterAllTensorFlowDialects(registry); mlir::MLIRContext context(registry); context.loadAllAvailableDialects(); auto module = TfToStablehlo(input_path, &context, exported_model_signatures, tag_names, input_arg_shapes_str, /*is_input_mlir_module=*/false); if (!module.ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_targets_test.cc
public: XlaLegalizeTargetsTest() : context_(GetDefaultDialectRegistry()), module_(mlir::ModuleOp::create(mlir::UnknownLoc::get(&context_))), builder_(&module_->getBodyRegion()) { context_.loadAllAvailableDialects(); } protected: mlir::MLIRContext context_; mlir::OwningOpRef<mlir::ModuleOp> module_; mlir::OpBuilder builder_; }; TEST_F(XlaLegalizeTargetsTest, CreatesConversionTargets) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 05:56:39 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/logging_hooks_test.cc
} 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/tfr/python/tfr_wrapper.cc
mlir::TF::TensorFlowDialect, mlir::func::FuncDialect, mlir::shape::ShapeDialect, mlir::TFR::TFRDialect>(); mlir::MLIRContext ctx(registry); ctx.loadAllAvailableDialects(); llvm::SourceMgr source_mgr = llvm::SourceMgr(); source_mgr.AddNewSourceBuffer(llvm::MemoryBuffer::getMemBuffer(input), llvm::SMLoc());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 13 06:54:12 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/cluster_tf_test.cc
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; mlir_module_ =
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
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; mlir_module_ =
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/api/v2/tf_dialect_to_executor_test.cc
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; mlir_module_ =
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/tensorflow/transforms/host_runtime/tpu_metadata_utils_test.cc
} class TpuMetadataUtilsTest : public ::testing::Test { public: TpuMetadataUtilsTest() { mlir::RegisterCommonToolingDialects(registry_); context_.appendDialectRegistry(registry_); context_.loadAllAvailableDialects(); } absl::StatusOr<std::vector<mlir::tf_device::ClusterFuncOp>> GetClusterFuncOps( absl::string_view mlir_module_filename) { TF_RETURN_IF_ERROR(CreateMlirModule(mlir_module_filename));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 12 04:22:33 UTC 2023 - 6.1K bytes - Viewed (0)