Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/tfrt/saved_model/tests/
Dsaved_model_test.cc50 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST_P() local
100 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
142 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
169 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
218 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
325 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
385 auto saved_model_dir = tensorflow::GetDataDependencyFilepath( in SetUp() local
484 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
528 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
613 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
[all …]
Dsaved_model_gpu_test.cc33 std::string saved_model_dir = tensorflow::GetDataDependencyFilepath( in TEST() local
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_utils.py27 def read_saved_model(saved_model_dir): argument
77 def get_saved_model_tag_sets(saved_model_dir): argument
94 def get_meta_graph_def(saved_model_dir, tag_set): argument
Dsaved_model_cli.py62 def _show_tag_sets(saved_model_dir): argument
76 def _show_signature_def_map_keys(saved_model_dir, tag_set): argument
138 def _show_inputs_outputs(saved_model_dir, tag_set, signature_def_key, indent=0): argument
177 def _show_defined_functions(saved_model_dir): argument
293 def _show_all(saved_model_dir): argument
316 def get_meta_graph_def(saved_model_dir, tag_set): argument
338 def get_signature_def_map(saved_model_dir, tag_set): argument
379 def run_saved_model_with_feed_dict(saved_model_dir, argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dinit_text_file_to_import.cc47 explicit InitTextFileToImportPass(std::string saved_model_dir) { in InitTextFileToImportPass()
59 StringRef saved_model_dir) in ConvertInitializeTableFromTextFileV2()
155 std::string saved_model_dir) { in CreateInitTextFileToImportPass()
Dfreeze_saved_model_assets.cc50 std::string saved_model_dir; member
115 std::string saved_model_dir) { in CreateFreezeAssetsPass()
/external/tensorflow/tensorflow/core/tfrt/saved_model/
Dsaved_model_testutil.cc72 TFRTSavedModelTest::TFRTSavedModelTest(const std::string& saved_model_dir) in TFRTSavedModelTest()
77 const std::string& saved_model_dir, in TFRTSavedModelTest()
93 void ComputeCurrentTFResult(const std::string& saved_model_dir, in ComputeCurrentTFResult()
136 void ComputeCurrentTFResult(const std::string& saved_model_dir, in ComputeCurrentTFResult()
198 GetWarmupRequests(absl::string_view saved_model_dir) { in GetWarmupRequests()
Dsaved_model.cc132 mlir::Operation* bound_input, absl::string_view saved_model_dir, in CreateTensorFromBoundInput()
148 absl::string_view saved_model_dir, mlir::ModuleOp module) { in GetFunctionSignaturesFromTFSavedModelMLIR()
305 const FallbackState& fallback_state, std::string saved_model_dir, in ImportSavedModel()
351 mlir::ModuleOp module, absl::string_view saved_model_dir) { in GetInitializersAndSignatures()
434 absl::string_view saved_model_dir, in ReadSavedModel()
455 Options options, absl::string_view saved_model_dir, in LoadSavedModel()
/external/tensorflow/tensorflow/lite/python/
Dconvert_saved_model.py45 def get_meta_graph_def(saved_model_dir, tag_set): argument
155 def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, argument
Dlite.py969 def _load_saved_model(self, saved_model_dir, saved_model_tags): argument
1153 saved_model_dir, argument
1749 def from_saved_model(cls, saved_model_dir, signature_keys=None, tags=None): argument
2181 saved_model_dir, argument
2711 saved_model_dir, argument
2839 saved_model_dir, argument
Dconvert_saved_model_test.py79 saved_model_dir, argument
Dutil_test.py266 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
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_tfl_passes.cc196 llvm::StringRef saved_model_dir, const toco::TocoFlags& toco_flags, in AddPostVariableFreezingTFToTFLConversionPasses()
380 void AddTFToTFLConversionPasses(llvm::StringRef saved_model_dir, in AddTFToTFLConversionPasses()
Dtf_to_tfl_flatbuffer.cc192 llvm::StringRef saved_model_dir, in ConvertTFExecutorToTFLOrFlatbuffer()
/external/tensorflow/tensorflow/core/tfrt/mla/
Dmla_utils.h32 inline bool IsMlarchive(absl::string_view saved_model_dir) { return false; } in IsMlarchive()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py376 def _RunGraphV1(self, saved_model_dir, inputs_data, config, num_runs=2): argument
402 def _RunGraphV2(self, saved_model_dir, inputs_data, graph_state, num_runs=2): argument
440 saved_model_dir, argument
465 def _CreateConverter(self, run_params, saved_model_dir, conversion_params): argument
490 def _GetCalibratedInferGraph(self, run_params, saved_model_dir, inputs_data): argument
547 def _GetInferGraph(self, run_params, saved_model_dir): argument
Dtrt_engine_op_shape_test.py54 def get_func_from_saved_model(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/lite/testing/
Dmlir_convert.py30 saved_model_dir, argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dtf_mlir_translate.cc153 absl::string_view saved_model_dir, in SavedModelObjectGraphToMlirImport()
177 absl::string_view saved_model_dir, in SavedModelSignatureDefsToMlirImport()
214 absl::string_view saved_model_dir, in SavedModelSignatureDefsToMlirImportLite()
/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/lite/experimental/mlir/testing/
Dmlir_convert.py190 def mlir_convert_saved_model(saved_model_dir, argument
/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

12