| /external/tensorflow/tensorflow/python/ |
| D | pywrap_mlir.py | 41 def experimental_convert_saved_model_to_mlir(saved_model_path, exported_names, argument
44 str(saved_model_path).encode('utf-8'),
48 def experimental_convert_saved_model_v1_to_mlir_lite(saved_model_path, tags, argument
52 str(saved_model_path).encode('utf-8'),
56 def experimental_convert_saved_model_v1_to_mlir(saved_model_path, tags, argument
60 str(saved_model_path).encode('utf-8'),
|
| D | mlir_wrapper.cc | 51 [](const std::string &saved_model_path, in PYBIND11_MODULE()
56 saved_model_path, exported_names, show_debug_info, status.get()); in PYBIND11_MODULE()
62 [](const std::string &saved_model_path, const std::string &tags, in PYBIND11_MODULE() argument
68 saved_model_path, tags, upgrade_legacy, show_debug_info, in PYBIND11_MODULE()
75 [](const std::string &saved_model_path, const std::string &tags, in PYBIND11_MODULE() argument
81 saved_model_path, tags, lift_variables, upgrade_legacy, in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saved_model_experimental.py | 66 saved_model_path, argument
136 saved_model_path,
139 _save_v1_format(model, saved_model_path, custom_objects, as_text,
143 _export_model_json(model, saved_model_path)
149 def _export_model_json(model, saved_model_path): argument
153 saved_model_utils.get_or_create_assets_dir(saved_model_path),
159 def _export_model_variables(model, saved_model_path): argument
161 saved_model_utils.get_or_create_variables_dir(saved_model_path)
162 checkpoint_prefix = saved_model_utils.get_variables_path(saved_model_path)
376 def load_from_saved_model(saved_model_path, custom_objects=None): argument
[all …]
|
| D | save_test.py | 721 saved_model_path = os.path.join(saved_model_dir, 'saved_model.h5')
730 with h5py.File(saved_model_path, 'w'):
731 keras.models.save_model(model, saved_model_path)
|
| /external/tensorflow/tensorflow/compiler/mlir/python/ |
| D | mlir.h | 57 const std::string &saved_model_path, const std::string &exported_names_str,
72 const std::string &saved_model_path, const std::string &tags,
86 const std::string &saved_model_path, const std::string &tags,
|
| D | mlir.cc | 137 const std::string &saved_model_path, const std::string &exported_names_str, in ExperimentalConvertSavedModelToMlir() argument
143 tensorflow::SavedModelV2Bundle::Load(saved_model_path, &bundle); in ExperimentalConvertSavedModelToMlir()
165 const std::string &saved_model_path, const std::string &tags, in ExperimentalConvertSavedModelV1ToMlirLite() argument
175 saved_model_path, tag_set, /*exported_names=*/{}, &context, in ExperimentalConvertSavedModelV1ToMlirLite()
186 const std::string &saved_model_path, const std::string &tags, in ExperimentalConvertSavedModelV1ToMlir() argument
196 tensorflow::LoadSavedModel({}, {}, saved_model_path, tag_set, &bundle); in ExperimentalConvertSavedModelV1ToMlir()
|
| /external/tensorflow/tensorflow/cc/saved_model/experimental/public/ |
| 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()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/debuginfo/ |
| D | saved_model_error.py | 48 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/lite/examples/experimental_new_converter/ |
| D | stack_trace_example.py | 60 saved_model_path = '/tmp/test.saved_model'
62 tf.saved_model.save(test_model, saved_model_path, options=save_options)
65 converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_path)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.experimental.pbtxt | 29 …argspec: "args=[\'model\', \'saved_model_path\', \'custom_objects\', \'as_text\', \'input_signatur…
33 …argspec: "args=[\'saved_model_path\', \'custom_objects\'], varargs=None, keywords=None, defaults=[…
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | freeze_test.py | 113 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/ |
| D | parallel_device_test.py | 317 saved_model_path = os.path.join(self.get_temp_dir(), "saved_model")
318 save.save(m, saved_model_path)
323 single_device_loaded = load.load(saved_model_path)
326 parallel_loaded = load.load(saved_model_path)
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert_test.py | 667 saved_model_path, converted_saved_model_path = _GetModelPaths(model_class)
668 original_def = _GetSignatureDef(saved_model_path)
675 _GetStructuredOutputs(saved_model_path),
|