Home
last modified time | relevance | path

Searched defs:cancel_mgr (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dcollective_param_resolver_distributed.cc36 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()
H A Dcancellable_call.h30 CancellableCall(CancellationManager* cancel_mgr, const string& remote_worker, in CancellableCall()
H A Dcollective_rma_distributed.cc47 CancellationManager* cancel_mgr, WorkerCacheInterface* wc) in RecvBufCall()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dtest_collective_executor_mgr.h44 CancellationManager* cancel_mgr, in CompleteParamsAsync()
51 CancellationManager* cancel_mgr, in CompleteGroupAsync()
58 CancellationManager* cancel_mgr, in CompleteInstanceAsync()
H A Dcollective_param_resolver_local_test.cc319 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
H A Dcollective_param_resolver_local.cc59 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()
H A Dbase_collective_executor.cc49 bool IsCancelled(CancellationManager* cancel_mgr) { in IsCancelled()
358 CancellationManager* cancel_mgr, StatusCallback done) { in CompleteParamsAsync()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcollective.h325 CancellationManager* cancel_mgr, in CompleteParamsAsync()
334 CancellationManager* cancel_mgr, in CompleteGroupAsync()
/aosp_15_r20/external/tensorflow/tensorflow/core/nccl/
H A Dcollective_communicator.cc96 CancellationManager* cancel_mgr = col_ctx->op_ctx->cancellation_manager(); in Enqueue() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfunction_ops.cc350 CancellationManager* cancel_mgr = nullptr; in ComputeAsync() local