Searched refs:asset_file_defs (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/cc/saved_model/ |
| D | loader.cc | 122 const std::vector<AssetFileDef>& asset_file_defs, in AddAssetsTensorsToInputs() argument
124 if (asset_file_defs.empty()) { in AddAssetsTensorsToInputs()
127 for (auto& asset_file_def : asset_file_defs) { in AddAssetsTensorsToInputs()
187 const std::vector<AssetFileDef>& asset_file_defs, in RunInitOp() argument
193 AddAssetsTensorsToInputs(export_dir, asset_file_defs, &inputs); in RunInitOp()
204 const std::vector<AssetFileDef>& asset_file_defs, in RunRestore() argument
231 AddAssetsTensorsToInputs(export_dir, asset_file_defs, &inputs); in RunRestore()
405 std::vector<AssetFileDef> asset_file_defs; in RestoreSession() local
406 TF_RETURN_IF_ERROR(internal::GetAssetFileDefs(meta_graph, &asset_file_defs)); in RestoreSession()
411 asset_file_defs, session->get())); in RestoreSession()
[all …]
|
| D | loader_util.cc | 64 std::vector<AssetFileDef>* asset_file_defs) { in GetAssetFileDefs() argument
69 asset_file_defs->push_back(asset); in GetAssetFileDefs()
84 asset_file_defs->push_back(asset_file_def); in GetAssetFileDefs()
|
| D | loader_util.h | 34 std::vector<AssetFileDef>* asset_file_defs);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 3522 std::vector<AssetFileDef> asset_file_defs; in ConvertAssets() local
3524 internal::GetAssetFileDefs(input_.meta_graph_def(), &asset_file_defs)); in ConvertAssets()
3527 results.reserve(asset_file_defs.size()); in ConvertAssets()
3531 for (const auto& asset : asset_file_defs) { in ConvertAssets()
|