| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_remote_worker.cc | 74 void GetStatusAsync(CallOptions* call_opts, const GetStatusRequest* request, in GetStatusAsync() argument
77 IssueRequest(request, response, getstatus_, std::move(done), call_opts, in GetStatusAsync()
87 void DeleteWorkerSessionAsync(CallOptions* call_opts, in DeleteWorkerSessionAsync() argument
92 call_opts); in DeleteWorkerSessionAsync()
107 void RunGraphAsync(CallOptions* call_opts, const RunGraphRequest* request, in RunGraphAsync() argument
109 IssueRequest(request, response, rungraph_, std::move(done), call_opts); in RunGraphAsync()
111 void RunGraphAsync(CallOptions* call_opts, RunGraphRequestWrapper* request, in RunGraphAsync() argument
115 rungraph_, std::move(done), call_opts); in RunGraphAsync()
130 void RecvBufAsync(CallOptions* call_opts, const RecvBufRequest* request, in RecvBufAsync() argument
172 IssueRequest(request, response, recvbuf_, callback, call_opts); in RecvBufAsync()
[all …]
|
| D | grpc_worker_service.cc | 243 CallOptions* call_opts = new CallOptions; in RunGraphHandler() local
248 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RunGraphHandler()
249 worker_->RunGraphAsync(call_opts, wrapped_request, wrapped_response, in RunGraphHandler()
250 [call, call_opts, wrapped_request, in RunGraphHandler()
257 delete call_opts; in RunGraphHandler()
269 CallOptions* call_opts = new CallOptions; in RecvTensorHandlerRaw() local
270 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RecvTensorHandlerRaw()
273 call_opts, &call->request, &call->response, in RecvTensorHandlerRaw()
274 [call, call_opts](const Status& s) { in RecvTensorHandlerRaw()
276 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, trace](const Status& status) { in RunStepHandler()
201 delete call_opts; in RunStepHandler()
258 CallOptions* call_opts = new CallOptions; in RunCallableHandler() local
262 call_opts->SetTimeout(default_session_config_.operation_timeout_in_ms()); in RunCallableHandler()
263 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RunCallableHandler()
[all …]
|
| D | grpc_state.h | 46 Response* response, StatusCallback done, CallOptions* call_opts,
50 stub, cq, method, request, response, std::move(done), call_opts,
87 (call_opts != nullptr ? call_opts->GetTimeout() : 0), max_retries,
94 Response* response, StatusCallback done, CallOptions* call_opts, in RPCState() argument
97 : call_opts_(call_opts), in RPCState()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/ |
| D | grpc_eager_service_impl.h | 76 auto call_opts = std::make_shared<CallOptions>(); in EnqueueHandler()
77 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in EnqueueHandler()
79 call_opts.get(), &call->request, &call->response))); in EnqueueHandler()
93 auto call_opts = std::make_shared<CallOptions>(); in RunComponentFunctionHandler()
94 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RunComponentFunctionHandler()
95 local_impl_.RunComponentFunction(call_opts.get(), &call->request, in RunComponentFunctionHandler()
97 [call, call_opts](const Status& s) { in RunComponentFunctionHandler()
|
| D | grpc_eager_client.cc | 143 void method##Async(CallOptions* call_opts, const method##Request* request, \
149 response, std::move(done_wrapped), call_opts, /*threadpool=*/nullptr, \
182 void StreamingEnqueueAsync(CallOptions* call_opts, in StreamingEnqueueAsync() argument
204 EnqueueAsync(call_opts, request, response, in StreamingEnqueueAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.cc | 54 auto call_opts = std::make_shared<CallOptions>(); in RunAsync() local
58 token, [call_opts, response, done]() { call_opts->StartCancel(); }); in RunAsync()
77 call_opts.get(), request_.get(), response.get(), in RunAsync()
78 [inputs, retvals, call_opts, response, device, in RunAsync()
|
| D | cluster_function_library_runtime.cc | 206 auto call_opts = std::make_shared<CallOptions>(); in Run() local
211 [call_opts, request, response, done]() { call_opts->StartCancel(); }); in Run()
223 call_opts.get(), request.get(), response.get(), in Run()
224 [request, response, rets, call_opts, cm, token, in Run()
|
| D | eager_client.h | 48 CallOptions* call_opts, const method##Request* request, \
67 virtual void StreamingEnqueueAsync(CallOptions* call_opts,
|
| D | eager_service_impl.cc | 429 CallOptions* call_opts, const RunComponentFunctionRequest* request, in RunComponentFunction() argument
481 call_opts->SetCancelCallback([cm] { cm->StartCancel(); }); in RunComponentFunction()
487 call_opts, response, eager_context, context, in RunComponentFunction()
489 call_opts->ClearCancelCallback(); in RunComponentFunction()
510 Status EagerServiceImpl::ExecuteOp(CallOptions* call_opts, in ExecuteOp() argument
521 if (call_opts) { in ExecuteOp()
523 call_opts->SetCancelCallback([cm] { cm->StartCancel(); }); in ExecuteOp()
549 Status EagerServiceImpl::Enqueue(CallOptions* call_opts, in Enqueue() argument
571 s = ExecuteOp(call_opts, item.operation(), context->Context(), &executor, in Enqueue()
|
| D | eager_service_impl.h | 93 Status Enqueue(CallOptions* call_opts, const EnqueueRequest* request,
100 void RunComponentFunction(CallOptions* call_opts,
211 Status ExecuteOp(CallOptions* call_opts, const Operation& operation,
|
| D | eager_service_impl_test.cc | 90 void EnqueueAsync(CallOptions* call_opts, const EnqueueRequest* request, in EnqueueAsync() argument
92 done(impl_->Enqueue(call_opts, request, response)); in EnqueueAsync()
95 void RunComponentFunctionAsync(CallOptions* call_opts, in RunComponentFunctionAsync() argument
99 impl_->RunComponentFunction(call_opts, request, response, std::move(done)); in RunComponentFunctionAsync()
102 void StreamingEnqueueAsync(CallOptions* call_opts, in StreamingEnqueueAsync() argument
528 CallOptions call_opts; in TestFunction() local
532 status = eager_service_impl.Enqueue(&call_opts, &remote_enqueue_request, in TestFunction()
540 call_opts.StartCancel(); in TestFunction()
621 CallOptions call_opts; in TestComponentFunction() local
624 eager_service_impl.RunComponentFunction(&call_opts, &run_comp_func_request, in TestComponentFunction()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_interface.h | 220 CallOptions call_opts; in CallAndWaitWithOptions() local
223 (this->*func)(&call_opts, req, resp, [&ret, &n](const Status& s) { in CallAndWaitWithOptions()
|
| D | master_session.cc | 211 CallOptions* call_opts, const RunCallableRequest& req,
324 int64 execution_count, PerStepState* pss, CallOptions* call_opts,
645 int64 execution_count, PerStepState* pss, CallOptions* call_opts, in RunPartitionsHelper() argument
746 call_opts->SetCancelCallback([&calls]() { in RunPartitionsHelper()
758 call_opts->ClearCancelCallback(); in RunPartitionsHelper()
809 PerStepState* pss, CallOptions* call_opts, const RunStepRequestWrapper& req, in RunPartitions() argument
829 call_opts, req, resp, cm, is_last_partial_run); in RunPartitions()
834 PerStepState* pss, CallOptions* call_opts, const RunCallableRequest& req, in RunPartitions() argument
855 call_opts, req, &wrapped_resp, cm, false /* is_last_partial_run */)); in RunPartitions()
1417 CallOptions call_opts; in DeleteWorkerSessions() member
[all …]
|