| 142 | } |
| 143 | |
| 144 | Status RunRestoreOp(const RunOptions& run_options, const StringPiece export_dir, |
| 145 | const std::vector<AssetFile>& asset_files, |
| 146 | const StringPiece restore_op_name, |
| 147 | const StringPiece variables_filename_const_op_name, |
| 148 | Session* session) { |
| 149 | LOG(INFO) << "Running restore op for SessionBundle: " << restore_op_name |
| 150 | << ", " << variables_filename_const_op_name; |
| 151 | Tensor variables_tensor = |
| 152 | CreateStringTensor(GetVariablesFilename(export_dir)); |
| 153 | std::vector<std::pair<string, Tensor>> inputs = { |
| 154 | {string(variables_filename_const_op_name), variables_tensor}}; |
| 155 | AddAssetsTensorsToInputs(export_dir, asset_files, &inputs); |
| 156 | RunMetadata run_metadata; |
| 157 | return session->Run(run_options, inputs, {}, {string(restore_op_name)}, |
| 158 | nullptr /* outputs */, &run_metadata); |
| 159 | } |
| 160 | |
| 161 | Status RunInitOp(const RunOptions& run_options, const StringPiece export_dir, |
| 162 | const std::vector<AssetFile>& asset_files, |
no test coverage detected