| 159 | } |
| 160 | |
| 161 | Status RunInitOp(const RunOptions& run_options, const StringPiece export_dir, |
| 162 | const std::vector<AssetFile>& asset_files, |
| 163 | const StringPiece init_op_name, Session* session) { |
| 164 | LOG(INFO) << "Running init op for SessionBundle"; |
| 165 | std::vector<std::pair<string, Tensor>> inputs; |
| 166 | AddAssetsTensorsToInputs(export_dir, asset_files, &inputs); |
| 167 | RunMetadata run_metadata; |
| 168 | return session->Run(run_options, inputs, {}, {string(init_op_name)}, |
| 169 | nullptr /* outputs */, &run_metadata); |
| 170 | } |
| 171 | |
| 172 | Status LoadSessionBundleFromPathUsingRunOptionsInternal( |
| 173 | const SessionOptions& options, const RunOptions& run_options, |
no test coverage detected