Searched defs:cancel_mgr (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.cc | 37 CancellationManager* cancel_mgr, in CompleteGroupCall()
61 bool is_source, CancellationManager* cancel_mgr, in CompleteInstanceCall()
109 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteParamsAsync()
133 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteGroupAsync()
169 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteInstanceAsync()
288 CancellationManager* cancel_mgr, const GroupRecCallback& done) { in CompleteGroupDistributed()
380 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteInstanceDistributed()
|
| D | cancellable_call.h | 30 CancellableCall(CancellationManager* cancel_mgr, const string& remote_worker, in CancellableCall()
|
| D | collective_rma_distributed.cc | 43 CancellationManager* cancel_mgr, WorkerCacheInterface* wc) in RecvBufCall()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | test_collective_executor_mgr.h | 40 CancellationManager* cancel_mgr, in CompleteParamsAsync()
47 CancellationManager* cancel_mgr, in CompleteGroupAsync()
54 CancellationManager* cancel_mgr, in CompleteInstanceAsync()
|
| D | collective_param_resolver_local_test.cc | 319 CancellationManager cancel_mgr; in TEST_F() local
345 CancellationManager cancel_mgr; in TEST_F() local
395 CancellationManager cancel_mgr; in TEST_F() local
450 CancellationManager cancel_mgr; in TEST_F() local
|
| D | collective_param_resolver_local.cc | 51 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteGroupAsync()
89 const GroupRecCallback& done, CancellationManager* cancel_mgr) { in CompleteGroupLocal()
641 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteParamsAsync()
658 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteInstanceAsync()
|
| D | base_collective_executor.cc | 48 bool IsCancelled(CancellationManager* cancel_mgr) { in IsCancelled()
348 CancellationManager* cancel_mgr, StatusCallback done) { in CompleteParamsAsync()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 84 CancellationManager* cancel_mgr = col_ctx->op_ctx->cancellation_manager(); in Enqueue() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 310 CancellationManager* cancel_mgr, in CompleteParamsAsync()
|