| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.cc | 37 CancellationManager* cancel_mgr, in CompleteGroupCall() argument
39 : CancellableCall(cancel_mgr, remote_worker, wc) { in CompleteGroupCall()
61 bool is_source, CancellationManager* cancel_mgr, in CompleteInstanceCall() argument
63 : CancellableCall(cancel_mgr, remote_worker, wc) { in CompleteInstanceCall()
109 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteParamsAsync() argument
113 device, cp, cancel_mgr, in CompleteParamsAsync()
114 [this, device, cp, cancel_mgr, done](Status s, const GroupRec* gr) { in CompleteParamsAsync()
124 CompleteInstanceDistributed(device.name(), gr, cp, cancel_mgr, done); in CompleteParamsAsync()
133 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteGroupAsync() argument
147 request->device_attributes(), cp, cancel_mgr, in CompleteGroupAsync()
[all …]
|
| D | collective_param_resolver_distributed.h | 38 CancellationManager* cancel_mgr,
43 CancellationManager* cancel_mgr,
48 CancellationManager* cancel_mgr,
69 CancellationManager* cancel_mgr,
87 CancellationManager* cancel_mgr,
|
| D | cancellable_call.h | 30 CancellableCall(CancellationManager* cancel_mgr, const string& remote_worker, in CancellableCall() argument
33 cancel_mgr_(cancel_mgr), in CancellableCall()
|
| D | collective_rma_distributed.cc | 43 CancellationManager* cancel_mgr, WorkerCacheInterface* wc) in RecvBufCall() argument
44 : CancellableCall(cancel_mgr, peer_task, wc) { in RecvBufCall()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local_test.cc | 319 CancellationManager cancel_mgr; in TEST_F() local
324 Env::Default()->SchedClosure([this, i, &cancel_mgr, &cp, &start, &done] { in TEST_F()
329 prl_->CompleteParamsAsync(GetDeviceAttributes(device), cp[i], &cancel_mgr, in TEST_F()
345 CancellationManager cancel_mgr; in TEST_F() local
354 &cancel_mgr, &cp, &done] { in TEST_F()
360 &cancel_mgr, in TEST_F()
373 Env::Default()->SchedClosure([this, group_key, instance_key, i, &cancel_mgr, in TEST_F()
379 prl_->CompleteParamsAsync(GetDeviceAttributes(device), cp[i], &cancel_mgr, in TEST_F()
395 CancellationManager cancel_mgr; in TEST_F() local
404 &cancel_mgr, &cp, &done] { in TEST_F()
[all …]
|
| D | collective_param_resolver_local.h | 51 CancellationManager* cancel_mgr,
56 CancellationManager* cancel_mgr,
61 CancellationManager* cancel_mgr,
89 CancellationManager* cancel_mgr)
|
| D | base_collective_executor.cc | 48 bool IsCancelled(CancellationManager* cancel_mgr) { in IsCancelled() argument
49 return cancel_mgr != nullptr && in IsCancelled()
50 (cancel_mgr->IsCancelled() || cancel_mgr->IsCancelling()); in IsCancelled()
348 CancellationManager* cancel_mgr, StatusCallback done) { in CompleteParamsAsync() argument
359 auto done_safe = [this, is_callback_called, cancel_mgr, trace_id, in CompleteParamsAsync()
364 if (!s.ok() && !IsCancelled(cancel_mgr)) { in CompleteParamsAsync()
388 cem_->GetParamResolver()->CompleteParamsAsync(device, cp, cancel_mgr, in CompleteParamsAsync()
|
| D | test_collective_executor_mgr.h | 40 CancellationManager* cancel_mgr, in CompleteParamsAsync() argument
47 CancellationManager* cancel_mgr, in CompleteGroupAsync() argument
54 CancellationManager* cancel_mgr, in CompleteInstanceAsync() argument
|
| D | collective_param_resolver_local.cc | 51 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteGroupAsync() argument
89 const GroupRecCallback& done, CancellationManager* cancel_mgr) { in CompleteGroupLocal() argument
98 if (cancel_mgr != nullptr) { in CompleteGroupLocal()
104 const CancellationToken token = cancel_mgr->get_cancellation_token(); in CompleteGroupLocal()
106 !cancel_mgr->RegisterCallback(token, [done, cancelled_mu, cancelled]() { in CompleteGroupLocal()
117 done_with_cleanup = [cancel_mgr, done, cancelled_mu, cancelled, token]( in CompleteGroupLocal()
121 if (*cancelled || !cancel_mgr->TryDeregisterCallback(token)) { in CompleteGroupLocal()
641 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteParamsAsync() argument
653 cancel_mgr); in CompleteParamsAsync()
658 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteInstanceAsync() argument
|
| D | base_collective_executor.h | 117 CancellationManager* cancel_mgr,
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 84 CancellationManager* cancel_mgr = col_ctx->op_ctx->cancellation_manager(); in Enqueue() local
85 if (cancel_mgr == nullptr) { in Enqueue()
88 CancellationToken cancel_token = cancel_mgr->get_cancellation_token(); in Enqueue()
90 !cancel_mgr->RegisterCallback(cancel_token, [this]() { in Enqueue()
98 participant->done_callback = [cancel_mgr, cancel_token, in Enqueue()
102 cancel_mgr->TryDeregisterCallback(cancel_token); in Enqueue()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 182 CancellationManager* cancel_mgr,
189 CancellationManager* cancel_mgr,
196 CancellationManager* cancel_mgr,
310 CancellationManager* cancel_mgr, in CompleteParamsAsync() argument
|