Home
last modified time | relevance | path

Searched refs:saved_model_path (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/python/
Dpywrap_mlir.py54 def experimental_convert_saved_model_to_mlir(saved_model_path, exported_names, argument
57 str(saved_model_path).encode('utf-8'),
61 def experimental_convert_saved_model_v1_to_mlir_lite(saved_model_path, argument
66 str(saved_model_path).encode('utf-8'),
71 def experimental_convert_saved_model_v1_to_mlir(saved_model_path, argument
76 str(saved_model_path).encode('utf-8'),
Dmlir_wrapper.cc66 [](const std::string &saved_model_path, in PYBIND11_MODULE()
71 saved_model_path, exported_names, show_debug_info, status.get()); in PYBIND11_MODULE()
77 [](const std::string &saved_model_path, in PYBIND11_MODULE() argument
84 saved_model_path, exported_names_str, tags, upgrade_legacy, in PYBIND11_MODULE()
91 [](const std::string &saved_model_path, in PYBIND11_MODULE() argument
98 saved_model_path, exported_names_str, tags, lift_variables, in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_experimental.py65 saved_model_path, argument
135 saved_model_path,
138 _save_v1_format(model, saved_model_path, custom_objects, as_text,
142 _export_model_json(model, saved_model_path)
148 def _export_model_json(model, saved_model_path): argument
152 _get_or_create_assets_dir(saved_model_path),
158 def _export_model_variables(model, saved_model_path): argument
160 _get_or_create_variables_dir(saved_model_path)
161 checkpoint_prefix = _get_variables_path(saved_model_path)
375 def load_from_saved_model(saved_model_path, custom_objects=None): argument
[all …]
/external/tensorflow/tensorflow/cc/saved_model/experimental/public/
Dsaved_model_api.h58 const std::string& saved_model_path, const Runtime& runtime,
103 const std::string& saved_model_path, const Runtime& runtime, Status* status, in Load() argument
109 TF_LoadSavedModel(saved_model_path.c_str(), runtime.GetTFEContext(), in Load()
118 saved_model_path.c_str(), runtime.GetTFEContext(), tags_vector.data(), in Load()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/debuginfo/
Dsaved_model_error.py48 saved_model_path = '/tmp/test.saved_model'
50 tf.saved_model.save(test_model, saved_model_path, options=save_options)
53 converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_path)
/external/tensorflow/tensorflow/compiler/mlir/python/
Dmlir.h67 const std::string &saved_model_path, const std::string &exported_names_str,
82 const std::string &saved_model_path, const std::string &exported_names_str,
97 const std::string &saved_model_path, const std::string &exported_names_str,
Dmlir.cc250 const std::string &saved_model_path, const std::string &exported_names_str, in ExperimentalConvertSavedModelToMlir() argument
256 tensorflow::SavedModelV2Bundle::Load(saved_model_path, &bundle); in ExperimentalConvertSavedModelToMlir()
278 const std::string &saved_model_path, const std::string &exported_names_str, in ExperimentalConvertSavedModelV1ToMlirLite() argument
291 saved_model_path, tag_set, absl::Span<std::string>(exported_names), in ExperimentalConvertSavedModelV1ToMlirLite()
302 const std::string &saved_model_path, const std::string &exported_names_str, in ExperimentalConvertSavedModelV1ToMlir() argument
312 tensorflow::LoadSavedModel({}, {}, saved_model_path, tag_set, &bundle); in ExperimentalConvertSavedModelV1ToMlir()
/external/tensorflow/tensorflow/lite/examples/experimental_new_converter/
Dstack_trace_example.py51 saved_model_path = '/tmp/test.saved_model'
53 tf.saved_model.save(test_model, saved_model_path, options=save_options)
56 converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_path)
/external/tensorflow/tensorflow/core/tfrt/saved_model/tests/
Dgen_pow_v2.py57 FLAGS.saved_model_path,
59 logging.info('Saved model to: %s', FLAGS.saved_model_path)
Dgen_error_v1.py43 shutil.rmtree(FLAGS.saved_model_path)
51 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
Dgen_pow.py44 shutil.rmtree(FLAGS.saved_model_path)
52 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
Dgen_saved_model_v2.py64 FLAGS.saved_model_path,
66 logging.info('Saved model to: %s', FLAGS.saved_model_path)
Dgen_resource_gather_v1.py44 shutil.rmtree(FLAGS.saved_model_path)
56 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
Dgen_while_v1.py46 shutil.rmtree(FLAGS.saved_model_path)
57 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
Dgen_sparse_tensor_input.py45 shutil.rmtree(FLAGS.saved_model_path)
56 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
Dgen_ref_type_tensor_input.py46 shutil.rmtree(FLAGS.saved_model_path)
61 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
Dgen_if_v1.py48 shutil.rmtree(FLAGS.saved_model_path)
63 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
Dgen_control_flow_v1.py45 shutil.rmtree(FLAGS.saved_model_path)
75 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
Dgen_hash_table_asset_v1.py57 shutil.rmtree(FLAGS.saved_model_path)
76 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
DBUILD398 "$(location gen_while_v1) --saved_model_path=$(RULEDIR)/while_v1",
412 "$(location gen_hash_table_asset_v1) --saved_model_path=$(RULEDIR)/hash_table_asset_v1",
425 "$(location gen_error_v1) --saved_model_path=$(RULEDIR)/error_v1",
438 "$(location gen_control_flow_v1) --saved_model_path=$(RULEDIR)/control_flow_v1",
451 "$(location gen_pow) --saved_model_path=$(RULEDIR)/pow",
464 "$(location gen_sparse_tensor_input) --saved_model_path=$(RULEDIR)/sparse_tensor_input",
476 cmd = "$(location gen_pow_v2) --saved_model_path=$(RULEDIR)/pow_v2",
Dgen_saved_model_v1.py46 shutil.rmtree(FLAGS.saved_model_path)
70 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
Dgen_dtype_coverage_v1.py43 shutil.rmtree(FLAGS.saved_model_path)
75 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
Dgen_saved_model.bzl15 "$(location " + script + ") --saved_model_path=$(RULEDIR)/" + model_name,
/external/tensorflow/tensorflow/examples/speech_commands/
Dfreeze_test.py113 saved_model_path = os.path.join(tmp_dir, 'saved_model')
128 freeze.save_saved_model(saved_model_path, sess, input_tensor,
/external/tensorflow/tensorflow/python/distribute/parallel_device/
Dparallel_device_test.py444 saved_model_path = os.path.join(self.get_temp_dir(), "saved_model")
445 save.save(m, saved_model_path)
450 single_device_loaded = load.load(saved_model_path)
455 parallel_loaded = load.load(saved_model_path)

12