| 148 | } |
| 149 | |
| 150 | Status RunIncrRestoreCheckpoint( |
| 151 | const RunOptions& run_options, |
| 152 | const std::string& full_ckpt_name, |
| 153 | const std::string& incr_ckpt_name, |
| 154 | const std::string& savedmodel_dir, |
| 155 | const StringPiece restore_op_name, |
| 156 | const StringPiece variable_filename_const_op_name, |
| 157 | const StringPiece incr_variable_filename_const_op_name, |
| 158 | const std::vector<AssetFileDef>& asset_file_defs, |
| 159 | Session* session, thread::ThreadPoolOptions thread_opt, |
| 160 | Session::CallableHandle** handler) { |
| 161 | LOG(INFO) << "Restoring checkpoint."; |
| 162 | // Find path to variables to be restored in export directory. |
| 163 | // Add variables to the graph. |
| 164 | Tensor variables_path_tensor(DT_STRING, TensorShape({})); |
| 165 | variables_path_tensor.scalar<string>()() = full_ckpt_name; |
| 166 | |
| 167 | std::vector<std::pair<string, Tensor>> inputs = { |
| 168 | {string(variable_filename_const_op_name), variables_path_tensor}}; |
| 169 | |
| 170 | Tensor incr_variables_path_tensor(DT_STRING, TensorShape({})); |
| 171 | incr_variables_path_tensor.scalar<string>()() = incr_ckpt_name; |
| 172 | inputs.push_back( |
| 173 | {string(incr_variable_filename_const_op_name), incr_variables_path_tensor}); |
| 174 | |
| 175 | util::AddAssetsTensorsToInputs(savedmodel_dir, asset_file_defs, &inputs); |
| 176 | |
| 177 | RunMetadata run_metadata; |
| 178 | return util::RunOnce(run_options, inputs, {}, {string(restore_op_name)}, |
| 179 | nullptr /* outputs */, &run_metadata, session, thread_opt, handler); |
| 180 | } |
| 181 | |
| 182 | Status RunRestoreCheckpoint( |
| 183 | const RunOptions& run_options, |
no test coverage detected