| 251 | } |
| 252 | |
| 253 | Status RecordPRunSetup(Session* session, |
| 254 | const std::vector<string>& input_names, |
| 255 | const std::vector<string>& output_names, |
| 256 | const std::vector<string>& target_nodes, |
| 257 | string* handle) { |
| 258 | ReplayOp op; |
| 259 | PartialRunSetupRequest* req = op.mutable_partial_run_setup(); |
| 260 | req->set_session_handle(SessionToHandle(session)); |
| 261 | for (auto& input : input_names) { |
| 262 | req->add_feed(input); |
| 263 | } |
| 264 | for (auto& output : output_names) { |
| 265 | req->add_fetch(output); |
| 266 | } |
| 267 | for (auto& target : target_nodes) { |
| 268 | req->add_target(target); |
| 269 | } |
| 270 | RUN_WITH_TIMESTAMP(PRunSetup, input_names, output_names, target_nodes, |
| 271 | handle); |
| 272 | op.mutable_partial_run_setup_response()->set_partial_run_handle(*handle); |
| 273 | return Flush(op); |
| 274 | } |
| 275 | |
| 276 | Status RecordPRun(Session* session, const string& handle, |
| 277 | const std::vector<std::pair<string, Tensor> >& inputs, |
nothing calls this directly
no test coverage detected