Home
last modified time | relevance | path

Searched defs:saved_model_dir (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/core/tfrt/saved_model/tests/
Dsaved_model_test.cc45 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST_P() local
94 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
141 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
170 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
277 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
337 auto saved_model_dir = tensorflow::GetDataDependencyFilepath( in SetUp() local
413 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
455 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
483 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
511 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
[all …]
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_utils.py31 def read_saved_model(saved_model_dir): argument
81 def get_saved_model_tag_sets(saved_model_dir): argument
98 def get_meta_graph_def(saved_model_dir, tag_set): argument
Dsaved_model_cli.py65 def _show_tag_sets(saved_model_dir): argument
79 def _show_signature_def_map_keys(saved_model_dir, tag_set): argument
141 def _show_inputs_outputs(saved_model_dir, tag_set, signature_def_key, indent=0): argument
180 def _show_defined_functions(saved_model_dir): argument
294 def _show_all(saved_model_dir): argument
317 def get_meta_graph_def(saved_model_dir, tag_set): argument
339 def get_signature_def_map(saved_model_dir, tag_set): argument
380 def run_saved_model_with_feed_dict(saved_model_dir, tag_set, signature_def_key, argument
/external/tensorflow/tensorflow/core/tfrt/saved_model/
Dsaved_model_testutil.cc60 TFRTSavedModelTest::TFRTSavedModelTest(const std::string& saved_model_dir) in TFRTSavedModelTest()
65 const std::string& saved_model_dir, in TFRTSavedModelTest()
81 void ComputeCurrentTFResult(const std::string& saved_model_dir, in ComputeCurrentTFResult()
124 void ComputeCurrentTFResult(const std::string& saved_model_dir, in ComputeCurrentTFResult()
Dsaved_model.cc166 mlir::Operation* bound_input, absl::string_view saved_model_dir, in CreateTensorFromBoundInput()
182 absl::string_view saved_model_dir, mlir::ModuleOp module) { in GetFunctionSignaturesFromTFSavedModelMLIR()
373 std::string saved_model_dir, bool import_user_signatures) { in ImportSavedModel()
417 mlir::ModuleOp module, absl::string_view saved_model_dir) { in GetInitializersAndSignatures()
470 Options options, absl::string_view saved_model_dir, in LoadSavedModel()
536 Options options, absl::string_view saved_model_dir, in LoadSavedModel()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dinit_text_file_to_import.cc45 explicit InitTextFileToImportPass(std::string saved_model_dir) { in InitTextFileToImportPass()
71 StringRef saved_model_dir) in ConvertInitializeTableFromTextFileV2()
167 std::string saved_model_dir) { in CreateInitTextFileToImportPass()
Dfreeze_saved_model_assets.cc55 std::string saved_model_dir; member
123 std::string saved_model_dir) { in CreateFreezeAssetsPass()
/external/tensorflow/tensorflow/lite/python/
Dconvert_saved_model.py49 def get_meta_graph_def(saved_model_dir, tag_set): argument
159 def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, argument
Dlite.py788 def _load_saved_model(self, saved_model_dir, saved_model_tags): argument
939 saved_model_dir, argument
1428 def from_saved_model(cls, saved_model_dir, signature_keys=None, tags=None): argument
1830 saved_model_dir, argument
2353 saved_model_dir, argument
2473 saved_model_dir, argument
Dconvert_saved_model_test.py83 saved_model_dir, argument
Dconvert.py517 saved_model_dir=None, argument
813 def convert_saved_model(saved_model_dir=None, argument
Dutil_test.py272 saved_model_dir=None, argument
/external/tensorflow/tensorflow/c/experimental/saved_model/internal/
Dsaved_model_api.cc41 std::string saved_model_dir(dirname); in TF_LoadSavedModel() local
66 std::string saved_model_dir(dirname); in TF_LoadSavedModelWithTags() local
Dsaved_model_api_test.cc49 std::string SavedModelPath(tensorflow::StringPiece saved_model_dir) { in SavedModelPath()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py369 def _RunGraphV1(self, saved_model_dir, inputs_data, config, num_runs=2): argument
395 def _RunGraphV2(self, saved_model_dir, inputs_data, graph_state, num_runs=2): argument
431 saved_model_dir, argument
454 def _CreateConverter(self, run_params, saved_model_dir, conversion_params): argument
479 def _GetCalibratedInferGraph(self, run_params, saved_model_dir, inputs_data): argument
510 def _GetInferGraph(self, run_params, saved_model_dir): argument
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Dasset.cc33 const std::string& saved_model_dir, in Create()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dtf_mlir_translate.cc149 absl::string_view saved_model_dir, in SavedModelObjectGraphToMlirImport()
169 absl::string_view saved_model_dir, in SavedModelSignatureDefsToMlirImport()
205 absl::string_view saved_model_dir, in SavedModelSignatureDefsToMlirImportLite()
/external/tensorflow/tensorflow/lite/experimental/mlir/testing/
Dmlir_convert.py190 def mlir_convert_saved_model(saved_model_dir, argument
/external/tensorflow/tensorflow/cc/saved_model/experimental/tests/
Dsaved_model_api_test.cc39 std::string SavedModelPath(tensorflow::StringPiece saved_model_dir) { in SavedModelPath()
/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
Drestore_ops_test.cc32 std::string CheckpointPrefix(StringPiece saved_model_dir) { in CheckpointPrefix()
/external/tensorflow/tensorflow/lite/toco/
Dmodel_flags.proto198 optional string saved_model_dir = 20; field
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dsaved_model_utils.cc204 const std::string& saved_model_dir, in LoadSavedAsset()
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants_test.py1571 def _addNoinlineAttributeToFunction(self, saved_model_dir, func_name): argument
/external/tensorflow/tensorflow/c/
Dc_api_test.cc1393 const string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
1469 const string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local