| 310 | } |
| 311 | |
| 312 | Status GrpcSession::PRunSetup(const std::vector<string>& input_names, |
| 313 | const std::vector<string>& output_names, |
| 314 | const std::vector<string>& target_nodes, |
| 315 | string* handle) { |
| 316 | // Convert to proto |
| 317 | PartialRunSetupRequest req; |
| 318 | PartialRunSetupResponse resp; |
| 319 | CallOptions call_options; |
| 320 | TF_RETURN_IF_ERROR(Handle(req.mutable_session_handle())); |
| 321 | for (const string& feed : input_names) { |
| 322 | req.add_feed(feed); |
| 323 | } |
| 324 | for (const string& fetch : output_names) { |
| 325 | req.add_fetch(fetch); |
| 326 | } |
| 327 | for (const string& target : target_nodes) { |
| 328 | req.add_target(target); |
| 329 | } |
| 330 | req.set_request_id(GetUniqueRequestId()); |
| 331 | call_options.SetTimeout(options_.config.operation_timeout_in_ms()); |
| 332 | TF_RETURN_IF_ERROR(master_->PartialRunSetup(&call_options, &req, &resp)); |
| 333 | *handle = resp.partial_run_handle(); |
| 334 | return Status::OK(); |
| 335 | } |
| 336 | |
| 337 | Status GrpcSession::PRun(const string& handle, |
| 338 | const std::vector<std::pair<string, Tensor>>& inputs, |
nothing calls this directly
no test coverage detected