Searched defs:cancel_mgr (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.cc | 36 CancellationManager* cancel_mgr, in CompleteGroupCall()
59 bool is_source, CancellationManager* cancel_mgr, in CompleteInstanceCall()
109 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteParamsAsync()
144 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteGroupAsync()
150 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteInstanceAsync()
275 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteGroupDistributed()
364 const string& device, CollectiveParams* cp, CancellationManager* cancel_mgr, in CompleteInstanceDistributed()
|
| D | cancellable_call.h | 30 CancellableCall(CancellationManager* cancel_mgr, const string& remote_worker, in CancellableCall()
|
| D | collective_rma_distributed.cc | 47 CancellationManager* cancel_mgr, WorkerCacheInterface* wc) in RecvBufCall()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | test_collective_executor_mgr.h | 44 CancellationManager* cancel_mgr, in CompleteParamsAsync()
51 CancellationManager* cancel_mgr, in CompleteGroupAsync()
58 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 | 59 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteGroupAsync()
132 CancellationManager* cancel_mgr, StatusCallback done) { in CompleteGroupLocal()
661 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteParamsAsync()
687 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteInstanceAsync()
|
| D | base_collective_executor.cc | 49 bool IsCancelled(CancellationManager* cancel_mgr) { in IsCancelled()
358 CancellationManager* cancel_mgr, StatusCallback done) { in CompleteParamsAsync()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 96 CancellationManager* cancel_mgr = col_ctx->op_ctx->cancellation_manager(); in Enqueue() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 325 CancellationManager* cancel_mgr, in CompleteParamsAsync()
334 CancellationManager* cancel_mgr, in CompleteGroupAsync()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 350 CancellationManager* cancel_mgr = nullptr; in ComputeAsync() local
|