/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/ |
H A D | quantize_model.py | 65 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 D | pywrap_quantize_model.cc | 58 [](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 D | quantize_model_wrapper.cc | 69 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 D | quantize_model.h | 29 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 D | quantize_model_wrapper.h | 30 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 D | quantize_model.cc | 112 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 D | saved_model_experimental.py | 64 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 D | pywrap_mlir.py | 50 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 D | mlir_wrapper.cc | 66 [](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 D | quantize_model_test_base.py | 216 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 D | mlir.cc | 220 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 D | mlir.h | 67 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 D | saved_model_api.h | 58 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 D | saved_model_error.py | 44 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 D | stack_trace_example.py | 47 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 D | BUILD | 472 "$(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 D | gen_pow_v2.py | 51 FLAGS.saved_model_path, 53 logging.info('Saved model to: %s', FLAGS.saved_model_path)
|
H A D | gen_error_v1.py | 37 shutil.rmtree(FLAGS.saved_model_path) 45 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
|
H A D | gen_data.py | 52 FLAGS.saved_model_path, 54 logging.info('Saved model to: %s', FLAGS.saved_model_path)
|
H A D | gen_pow.py | 38 shutil.rmtree(FLAGS.saved_model_path) 46 sm_builder = builder.SavedModelBuilder(FLAGS.saved_model_path)
|
H A D | gen_matmul_gpu.py | 62 FLAGS.saved_model_path, 64 logging.info('Saved model to: %s', FLAGS.saved_model_path)
|
H A D | gen_saved_model_v2.py | 58 FLAGS.saved_model_path, 60 logging.info('Saved model to: %s', FLAGS.saved_model_path)
|
H A D | gen_variable_on_tpu.py | 62 FLAGS.saved_model_path, 64 logging.info('Saved model to: %s', FLAGS.saved_model_path)
|
H A D | gen_resource_gather_v1.py | 38 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 D | base_delegate_test.py | 74 saved_model_path = os.path.join(test_dir, "saved_model") 75 save.save(a, saved_model_path) 77 loaded = load.load(saved_model_path)
|