Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/
H A Dquantize_model.py65 def _is_qat_saved_model(saved_model_path: str):
67 saved_model_proto = saved_model_loader.parse_saved_model(saved_model_path)
78 def _get_signatures_from_saved_model(saved_model_path: str,
82 loader = saved_model_loader.SavedModelLoader(saved_model_path)
535 saved_model_path: str, signature_def_keys: Sequence[str],
553 quantize_model_wrapper.quantize_qat_model(saved_model_path,
565 saved_model_path: str,
600 saved_model_path, ','.join(signature_def_keys), ','.join(tags)))
712 saved_model_path: str,
751 is_qat_saved_model = _is_qat_saved_model(saved_model_path)
[all …]
H A Dpywrap_quantize_model.cc58 [](const absl::string_view saved_model_path, in PYBIND11_MODULE()
63 tensorflow::quantization::QuantizeQatModel(saved_model_path, in PYBIND11_MODULE()
72 [](const absl::string_view saved_model_path, in PYBIND11_MODULE() argument
78 saved_model_path, exported_names_str, tags, in PYBIND11_MODULE()
86 [](const absl::string_view saved_model_path, in PYBIND11_MODULE() argument
91 saved_model_path, exported_names_str, tags)); in PYBIND11_MODULE()
98 [](const absl::string_view saved_model_path, in PYBIND11_MODULE() argument
104 saved_model_path, exported_names_str, tags, in PYBIND11_MODULE()
H A Dquantize_model_wrapper.cc69 PyObject* QuantizeQatModel(const absl::string_view saved_model_path, in QuantizeQatModel() argument
74 saved_model_path, exported_names_str, tags, quant_opts_serialized); in QuantizeQatModel()
85 const absl::string_view saved_model_path, in QuantizePtqDynamicRange() argument
89 saved_model_path, exported_names_str, tags, quant_opts_serialized); in QuantizePtqDynamicRange()
102 const absl::string_view saved_model_path, in QuantizePtqModelPreCalibration() argument
105 internal::QuantizePtqModelPreCalibration(saved_model_path, in QuantizePtqModelPreCalibration()
118 const absl::string_view saved_model_path, in QuantizePtqModelPostCalibration() argument
123 saved_model_path, exported_names_str, tags, quant_opts_serialized); in QuantizePtqModelPostCalibration()
H A Dquantize_model.h29 absl::string_view saved_model_path, absl::string_view exported_names_str,
34 absl::string_view saved_model_path, absl::string_view exported_names_str,
38 absl::string_view saved_model_path, absl::string_view exported_names_str,
42 absl::string_view saved_model_path, absl::string_view exported_names_str,
H A Dquantize_model_wrapper.h30 PyObject* QuantizeQatModel(absl::string_view saved_model_path,
35 PyObject* QuantizePtqDynamicRange(absl::string_view saved_model_path,
40 PyObject* QuantizePtqModelPreCalibration(absl::string_view saved_model_path,
45 absl::string_view saved_model_path, absl::string_view exported_names_str,
H A Dquantize_model.cc112 const absl::string_view saved_model_path, in QuantizeQatModel() argument
141 SavedModelSignatureDefsToMlirImport(saved_model_path, tag_set, in QuantizeQatModel()
208 const absl::string_view saved_model_path, in QuantizePtqModelPreCalibration() argument
229 SavedModelSignatureDefsToMlirImport(saved_model_path, tag_set, in QuantizePtqModelPreCalibration()
271 const absl::string_view saved_model_path, in QuantizePtqModelPostCalibration() argument
300 SavedModelSignatureDefsToMlirImport(saved_model_path, tag_set, in QuantizePtqModelPostCalibration()
354 const absl::string_view saved_model_path, in QuantizePtqDynamicRange() argument
383 SavedModelSignatureDefsToMlirImport(saved_model_path, tag_set, in QuantizePtqDynamicRange()
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/saving/
H A Dsaved_model_experimental.py64 saved_model_path, argument
134 saved_model_path,
137 _save_v1_format(model, saved_model_path, custom_objects, as_text,
141 _export_model_json(model, saved_model_path)
147 def _export_model_json(model, saved_model_path): argument
151 _get_or_create_assets_dir(saved_model_path),
157 def _export_model_variables(model, saved_model_path): argument
159 _get_or_create_variables_dir(saved_model_path)
160 checkpoint_prefix = _get_variables_path(saved_model_path)
374 def load_from_saved_model(saved_model_path, custom_objects=None): argument
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dpywrap_mlir.py50 def experimental_convert_saved_model_to_mlir(saved_model_path, exported_names, argument
53 str(saved_model_path).encode('utf-8'),
57 def experimental_convert_saved_model_v1_to_mlir_lite(saved_model_path, argument
62 str(saved_model_path).encode('utf-8'),
67 def experimental_convert_saved_model_v1_to_mlir(saved_model_path, argument
72 str(saved_model_path).encode('utf-8'),
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/
H A Dquantize_model_test_base.py216 def _save_tf1_model(self, sess: session.Session, saved_model_path: str,
231 v1_builder = builder.SavedModelBuilder(saved_model_path)
240 saved_model_path: str,
272 saved_model_path,
399 saved_model_path: str,
431 saved_model_path,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/python/
H A Dmlir.cc220 const std::string& saved_model_path, const std::string& exported_names_str, in ExperimentalConvertSavedModelToMlir() argument
226 tensorflow::SavedModelV2Bundle::Load(saved_model_path, &bundle); in ExperimentalConvertSavedModelToMlir()
248 const std::string& saved_model_path, const std::string& exported_names_str, in ExperimentalConvertSavedModelV1ToMlirLite() argument
261 saved_model_path, tag_set, absl::Span<std::string>(exported_names), in ExperimentalConvertSavedModelV1ToMlirLite()
272 const std::string& saved_model_path, const std::string& exported_names_str, in ExperimentalConvertSavedModelV1ToMlir() argument
282 tensorflow::LoadSavedModel({}, {}, saved_model_path, tag_set, &bundle); in ExperimentalConvertSavedModelV1ToMlir()
H A 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,
/aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/experimental/public/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/debuginfo/
H A Dsaved_model_error.py44 saved_model_path = '/tmp/test.saved_model'
46 tf.saved_model.save(test_model, saved_model_path, options=save_options)
49 converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_path)
/aosp_15_r20/external/tensorflow/tensorflow/lite/examples/experimental_new_converter/
H A Dstack_trace_example.py47 saved_model_path = '/tmp/test.saved_model'
49 tf.saved_model.save(test_model, saved_model_path, options=save_options)
52 converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_path)
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/tests/
H A DBUILD472 "$(location gen_while_v1) --saved_model_path=$(RULEDIR)/while_v1",
486 "$(location gen_hash_table_asset_v1) --saved_model_path=$(RULEDIR)/hash_table_asset_v1",
499 "$(location gen_error_v1) --saved_model_path=$(RULEDIR)/error_v1",
512 "$(location gen_control_flow_v1) --saved_model_path=$(RULEDIR)/control_flow_v1",
525 "$(location gen_pow) --saved_model_path=$(RULEDIR)/pow",
538 "$(location gen_sparse_tensor_input) --saved_model_path=$(RULEDIR)/sparse_tensor_input",
550 cmd = "$(location gen_pow_v2) --saved_model_path=$(RULEDIR)/pow_v2",
560 cmd = "$(location gen_data) --saved_model_path=$(RULEDIR)/data",
H A Dgen_pow_v2.py51 FLAGS.saved_model_path,
53 logging.info('Saved model to: %s', FLAGS.saved_model_path)
H A Dgen_error_v1.py37 shutil.rmtree(FLAGS.saved_model_path)
45 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
H A Dgen_data.py52 FLAGS.saved_model_path,
54 logging.info('Saved model to: %s', FLAGS.saved_model_path)
H A Dgen_pow.py38 shutil.rmtree(FLAGS.saved_model_path)
46 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
H A Dgen_matmul_gpu.py62 FLAGS.saved_model_path,
64 logging.info('Saved model to: %s', FLAGS.saved_model_path)
H A Dgen_saved_model_v2.py58 FLAGS.saved_model_path,
60 logging.info('Saved model to: %s', FLAGS.saved_model_path)
H A Dgen_variable_on_tpu.py62 FLAGS.saved_model_path,
64 logging.info('Saved model to: %s', FLAGS.saved_model_path)
H A Dgen_resource_gather_v1.py38 shutil.rmtree(FLAGS.saved_model_path)
50 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
/aosp_15_r20/external/tensorflow/tensorflow/python/trackable/
H A Dbase_delegate_test.py74 saved_model_path = os.path.join(test_dir, "saved_model")
75 save.save(a, saved_model_path)
77 loaded = load.load(saved_model_path)

12