- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for module_ctor (0.18 sec)
-
tensorflow/cc/saved_model/testdata/generate_saved_models.py
print("Allowed ModuleNames:", MODULE_CTORS.keys()) return 1 _, export_path, module_name = args module_ctor, version = MODULE_CTORS.get(module_name) if not module_ctor: print("Expected ModuleName to be one of:", MODULE_CTORS.keys()) return 2 os.makedirs(export_path) tf_module = module_ctor() if version == 2:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Sep 18 18:06:18 UTC 2023 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf_mlir_translate_main.cc
if (import_saved_model_object_graph) { mlir::MLIRContext context; auto module_or = tensorflow::SavedModelObjectGraphToMlirImport( input_filename, tags, exported_names, &context); if (!module_or.status().ok()) return 1; module_or.value()->print(output->os()); } else if (import_saved_model_signature_defs) { mlir::MLIRContext context;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir_test.cc
func.return %0 : tensor<1xi32> } })"; absl::StatusOr<std::string> CompileMlirModule(bool compile_to_xla_hlo, const char* module_str) { MlirToHloArgs mlir_to_hlo_args; mlir_to_hlo_args.mlir_module = module_str; std::vector<TensorShape> arg_shapes; TPUCompileMetadataProto metadata_proto; bool use_tuple_args = true; std::vector<ShardingAndIndex> arg_core_mapping;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util_test.cc
if (!status.ok()) { return status; } return mlir_module; } TEST(XlaRewriteUtilTest, TestEraseClusterFuncs) { static const char* const module_str = R"( module attributes {tf.devices = ["/job:worker/replica:0/task:0/device:CPU:0", "/job:worker/replica:0/task:0/device:GPU:0"]} { func.func @convert_cluster_func(%arg0: tensor<i32>) -> () {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc
enable_soft_placement, set_original_tf_func_name}; auto module_or = tensorflow::GraphdefToMlirTranslateFunction( input, input_arrays, input_dtypes, input_shapes, output_arrays, control_output_arrays, options, context); if (!module_or.status().ok()) return nullptr; return std::move(module_or).value(); } static TranslateToMLIRRegistration GraphdefToMlirTranslate(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/mlir/mlir_transform.cc
->cast<mlir::MLIRContext*>(); // Load the saved model into MLIR TF dialect. absl::Span<std::string> exported_names(nullptr, 0); auto module_or = tensorflow::ConvertSavedModelToMlir(&bundle, context, exported_names); if (!module_or.status().ok()) { return None(); } // Make a module to wrap MLIR module and allow getting strings and running // transforms.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jul 12 16:28:19 UTC 2022 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_to_hlo_test.cc
} })"; absl::StatusOr<XlaCompiler::CompilationResult> CompileMlirModule( const char* module_str) { MlirToHloArgs mlir_to_hlo_args; mlir_to_hlo_args.rollout_state = ConfigProto::Experimental::MLIR_BRIDGE_ROLLOUT_UNSPECIFIED; mlir_to_hlo_args.mlir_module = module_str; se::Platform* platform = se::PlatformManager::PlatformWithName("Host").value(); auto client =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 6K bytes - Viewed (0)