/aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/ |
H A D | saved_model_bundle_test.cc | 65 void ValidateAssets(const string& export_dir, in ValidateAssets() 82 void CheckSavedModelBundle(const string& export_dir, in CheckSavedModelBundle() 119 const string export_dir = in TEST_F() local 133 const string export_dir = in TEST_F() local 145 const string export_dir = in TEST_F() local 161 const string export_dir = in TEST_F() local 182 const string export_dir = in TEST_F() local 198 const string export_dir = in TEST_F() local 218 const string export_dir = in TEST_F() local 232 const string export_dir = in TEST_F() local [all …]
|
H A D | saved_model_bundle_lite_test.cc | 54 void ValidateAssets(const string& export_dir, in ValidateAssets() 71 void CheckSavedModelBundle(const string& export_dir, in CheckSavedModelBundle() 117 const string export_dir = in TEST_F() local 131 const string export_dir = in TEST_F() local 144 const string export_dir = in TEST_F() local 156 const string export_dir = in TEST_F() local 172 const string export_dir = in TEST_F() local 192 const string export_dir = in TEST_F() local 206 const string export_dir = in TEST_F() local 218 const string export_dir = in TEST_F() local [all …]
|
H A D | reader_test.cc | 60 const string export_dir = GetDataDependencyFilepath(TestDataSharded()); in TEST_F() local 69 const string export_dir = GetDataDependencyFilepath(TestDataSharded()); in TEST_F() local 82 const string export_dir = GetDataDependencyFilepath(TestDataSharded()); in TEST_F() local 95 const string export_dir = GetDataDependencyFilepath(TestDataPbTxt()); in TEST_F() local 104 const string export_dir = GetDataDependencyFilepath("missing-path"); in TEST_F() local 111 const string export_dir = GetDataDependencyFilepath(TestDataSharded()); in TEST_F() local 121 const string export_dir = GetDataDependencyFilepath("missing-path"); in TEST_F() local 134 const string export_dir = GetDataDependencyFilepath(TestDataSharded()); in TEST_F() local
|
H A D | fingerprinting_test.cc | 65 const std::string export_dir = in TEST() local 85 const std::string export_dir = io::JoinPath( in TEST() local 110 const std::string export_dir = io::JoinPath( in TEST() local 134 const std::string export_dir = io::JoinPath( in TEST() local 148 const std::string export_dir = io::JoinPath( in TEST() local 158 const std::string export_dir = io::JoinPath( in TEST() local
|
H A D | loader.cc | 144 void AddAssetsTensorsToInputs(const StringPiece export_dir, in AddAssetsTensorsToInputs() 208 Status RunInitOp(const RunOptions& run_options, const string& export_dir, in RunInitOp() 224 Status RunRestore(const RunOptions& run_options, const string& export_dir, in RunRestore() 280 const string& export_dir, in LoadSavedModelInternal() 295 const RunOptions& run_options, const string& export_dir, in LoadSavedModel() 430 const MetaGraphDef& meta_graph, const string& export_dir, in RestoreSession() 461 const RunOptions& run_options, const string& export_dir, in LoadSavedModel() 485 bool MaybeSavedModelDirectory(const string& export_dir) { in MaybeSavedModelDirectory()
|
H A D | reader.cc | 43 Status ReadSavedModel(absl::string_view export_dir, in ReadSavedModel() 117 Status ReadMetaGraphDefFromSavedModel(const string& export_dir, in ReadMetaGraphDefFromSavedModel() 128 const string& export_dir, in ReadSavedModelDebugInfoIfPresent()
|
H A D | bundle_v2_test.cc | 74 const string export_dir = io::JoinPath( in TEST_F() local 87 const string export_dir = in TEST_F() local 104 const string export_dir = io::JoinPath( in TEST_F() local
|
H A D | bundle_v2.cc | 39 Status ReadSavedModelProto(const string& export_dir, in ReadSavedModelProto() 99 Status SavedModelV2Bundle::Load(const std::string& export_dir, in Load()
|
H A D | loader_util.cc | 31 Status GetInitOp(const string& export_dir, const MetaGraphDef& meta_graph_def, in GetInitOp()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/ |
H A D | utils_impl.py | 216 def get_or_create_variables_dir(export_dir): argument 223 def get_variables_dir(export_dir): argument 229 def get_variables_path(export_dir): argument 236 def get_or_create_assets_dir(export_dir): argument 245 def get_assets_dir(export_dir): argument 251 def get_or_create_debug_dir(export_dir): argument 260 def get_saved_model_pbtxt_path(export_dir): argument 266 def get_saved_model_pb_path(export_dir): argument 272 def get_debug_dir(export_dir): argument
|
H A D | loader_impl.py | 44 def parse_saved_model_with_debug_info(export_dir): argument 74 def parse_saved_model(export_dir): argument 121 def get_asset_tensors(export_dir, meta_graph_def_to_load, import_scope=None): argument 229 def maybe_saved_model_directory(export_dir): argument 250 def contains_saved_model(export_dir): argument 274 def load(sess, tags, export_dir, import_scope=None, **saver_kwargs): argument 343 def __init__(self, export_dir): argument 355 def export_dir(self): member in SavedModelLoader
|
H A D | load.py | 145 def __init__(self, object_graph_proto, saved_model_proto, export_dir, argument 723 def load(export_dir, tags=None, options=None): argument 820 def load_partial(export_dir, filters, tags=None, options=None): argument
|
H A D | save.py | 1018 def _export_debug_info(exported_graph, export_dir): argument 1050 def save(obj, export_dir, signatures=None, options=None): argument 1238 export_dir, argument
|
H A D | saved_model_test.py | 104 export_dir, argument 1377 export_dir, argument 1441 def _testInitOpsWithNonEmptyCollection(self, export_dir, key): argument
|
H A D | builder_impl.py | 93 def __init__(self, export_dir): argument 438 def __init__(self, export_dir): argument
|
H A D | simple_save.py | 31 def simple_save(session, export_dir, inputs, outputs, legacy_init_op=None): argument
|
H A D | pywrap_saved_model.cc | 33 m.def("Save", [](const char* export_dir) { in PYBIND11_MODULE()
|
H A D | pywrap_saved_model_fingerprinting.cc | 36 [](std::string serialized_saved_model, std::string export_dir) { in DefineFingerprintingModule()
|
H A D | method_name_updater.py | 56 def __init__(self, export_dir): argument
|
H A D | load_v1_in_v2.py | 280 def load(export_dir, tags): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/saving/ |
H A D | saved_model_experimental.py | 435 def _get_or_create_variables_dir(export_dir): argument 442 def _get_variables_dir(export_dir): argument 449 def _get_variables_path(export_dir): argument 456 def _get_or_create_assets_dir(export_dir): argument 465 def _get_assets_dir(export_dir): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/cc/experimental/libexport/ |
H A D | save.cc | 22 Status Save(const std::string& export_dir) { in Save()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/ |
H A D | trt_convert_test.py | 690 def _GetSignatureDef(export_dir): argument 711 def _GetStructuredOutputs(export_dir): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/python/trackable/ |
H A D | asset.py | 92 def _deserialize_from_proto(cls, object_proto, export_dir, asset_file_def, argument
|
/aosp_15_r20/external/tensorflow/tensorflow/java/src/main/native/ |
H A D | saved_model_bundle_jni.cc | 24 JNIEnv* env, jclass clazz, jstring export_dir, jobjectArray tags, in Java_org_tensorflow_SavedModelBundle_load()
|