| 420 | } |
| 421 | |
| 422 | Status GrpcSession::MakeCallable(const CallableOptions& callable_options, |
| 423 | CallableHandle* out_handle) { |
| 424 | MakeCallableRequest req; |
| 425 | TF_RETURN_IF_ERROR(Handle(req.mutable_session_handle())); |
| 426 | *req.mutable_options() = callable_options; |
| 427 | req.set_request_id(GetUniqueRequestId()); |
| 428 | MakeCallableResponse resp; |
| 429 | CallOptions call_options; |
| 430 | call_options.SetTimeout(options_.config.operation_timeout_in_ms()); |
| 431 | TF_RETURN_IF_ERROR(master_->MakeCallable(&call_options, &req, &resp)); |
| 432 | *out_handle = resp.handle(); |
| 433 | return Status::OK(); |
| 434 | } |
| 435 | |
| 436 | Status GrpcSession::RunCallable(CallableHandle handle, |
| 437 | const std::vector<Tensor>& feed_tensors, |