| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 226 CallOptions* call_opts = new CallOptions; in RunGraphHandler() local
231 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RunGraphHandler()
232 auto done_cb = [call, call_opts, wrapped_request, in RunGraphHandler()
239 delete call_opts; in RunGraphHandler()
245 auto compute_fn = [this, call_opts, wrapped_request, in RunGraphHandler()
247 worker_->RunGraphAsync(call_opts, wrapped_request, wrapped_response, in RunGraphHandler()
265 CallOptions* call_opts = new CallOptions; in RecvTensorHandlerRaw() local
266 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RecvTensorHandlerRaw()
268 auto done_cb = [call, call_opts](const Status& s) { in RecvTensorHandlerRaw()
270 delete call_opts; in RecvTensorHandlerRaw()
[all …]
|
| D | grpc_master_service.cc | 186 CallOptions* call_opts = new CallOptions; in RunStepHandler() local
188 call_opts->SetTimeout(call->request.options().timeout_in_ms()); in RunStepHandler()
190 call_opts->SetTimeout(default_session_config_.operation_timeout_in_ms()); in RunStepHandler()
196 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RunStepHandler()
198 call_opts, wrapped_request, wrapped_response, in RunStepHandler()
199 [call, call_opts, wrapped_request, wrapped_response, in RunStepHandler()
202 delete call_opts; in RunStepHandler()
259 CallOptions* call_opts = new CallOptions; in RunCallableHandler() local
263 call_opts->SetTimeout(default_session_config_.operation_timeout_in_ms()); in RunCallableHandler()
264 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RunCallableHandler()
[all …]
|
| D | grpc_remote_worker.cc | 85 void DeleteWorkerSessionAsync(CallOptions* call_opts, in DeleteWorkerSessionAsync() argument
90 call_opts); in DeleteWorkerSessionAsync()
105 void RunGraphAsync(CallOptions* call_opts, const RunGraphRequest* request, in RunGraphAsync() argument
107 IssueRequest(request, response, rungraph_, std::move(done), call_opts); in RunGraphAsync()
109 void RunGraphAsync(CallOptions* call_opts, RunGraphRequestWrapper* request, in RunGraphAsync() argument
113 rungraph_, std::move(done), call_opts); in RunGraphAsync()
128 void RecvBufAsync(CallOptions* call_opts, const RecvBufRequest* request, in RecvBufAsync() argument
167 IssueRequest(request, response, recvbuf_, *cb_to_use, call_opts); in RecvBufAsync()
170 void CompleteGroupAsync(CallOptions* call_opts, in CompleteGroupAsync() argument
174 IssueRequest(request, response, completegroup_, std::move(done), call_opts); in CompleteGroupAsync()
[all …]
|
| D | grpc_state.h | 44 Response* response, StatusCallback done, CallOptions* call_opts,
47 call_opts, threadpool, /*fail_fast=*/false,
53 Response* response, StatusCallback done, CallOptions* call_opts, in RPCState() argument
56 : call_opts_(call_opts), in RPCState()
|
| D | grpc_rpc_factory.cc | 60 CallOptions* call_opts() { return &call_opts_; } in call_opts() function in tensorflow::internal::GrpcCall
212 /*done=*/[call](const Status& s) { call->Done(s); }, call->call_opts(), in StartCall()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_interface.h | 212 CallOptions call_opts; in CallAndWaitWithOptions() local
215 (this->*func)(&call_opts, req, resp, [&ret, &n](const Status& s) { in CallAndWaitWithOptions()
|
| D | master_session.cc | 207 CallOptions* call_opts, const RunCallableRequest& req,
320 int64 execution_count, PerStepState* pss, CallOptions* call_opts,
594 int64 execution_count, PerStepState* pss, CallOptions* call_opts, in RunPartitionsHelper() argument
693 call_opts->SetCancelCallback([&calls]() { calls.StartCancel(); }); in RunPartitionsHelper()
701 call_opts->ClearCancelCallback(); in RunPartitionsHelper()
752 PerStepState* pss, CallOptions* call_opts, const RunStepRequestWrapper& req, in RunPartitions() argument
772 call_opts, req, resp, cm, is_last_partial_run); in RunPartitions()
777 PerStepState* pss, CallOptions* call_opts, const RunCallableRequest& req, in RunPartitions() argument
798 call_opts, req, &wrapped_resp, cm, false /* is_last_partial_run */)); in RunPartitions()
1325 CallOptions call_opts; in DeleteWorkerSessions() member
[all …]
|