experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Enable saved model restoration using just a .pb file.

Browse Source 
PiperOrigin-RevId: 338342231
Change-Id: Idfb9a0692fb414d8bc153caf031fe9987fa83b30
This commit is contained in:
A. Unique TensorFlower 2020-10-21 14:36:04 -07:00 committed by TensorFlower Gardener
parent 7912c72005
commit 47f21101fa
1 changed files with 6 additions and 4 deletions

2
tensorflow/cc/saved_model/loader.cc
View File

@ -404,10 +404,12 @@ Status RestoreSession(const RunOptions& run_options,
const uint64 read_start_microseconds = Env::Default()->NowMicros(); const uint64 read_start_microseconds = Env::Default()->NowMicros();
std::vector<AssetFileDef> asset_file_defs; std::vector<AssetFileDef> asset_file_defs;
TF_RETURN_IF_ERROR(internal::GetAssetFileDefs(meta_graph, &asset_file_defs)); TF_RETURN_IF_ERROR(internal::GetAssetFileDefs(meta_graph, &asset_file_defs));
if (meta_graph.has_saver_def()) {
TF_RETURN_IF_ERROR(RunRestore(run_options, export_dir, TF_RETURN_IF_ERROR(RunRestore(run_options, export_dir,
meta_graph.saver_def().restore_op_name(), meta_graph.saver_def().restore_op_name(),
meta_graph.saver_def().filename_tensor_name(), meta_graph.saver_def().filename_tensor_name(),
asset_file_defs, session->get())); asset_file_defs, session->get()));
}
// Record walltime spent in restoring graph from disk, but postpone metric // Record walltime spent in restoring graph from disk, but postpone metric
// increments until graph init finishes. // increments until graph init finishes.
const uint64 restore_graph_walltime = const uint64 restore_graph_walltime =