Home
last modified time | relevance | path

Searched refs:cancel_mgr_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dcancellable_call.h32 : cancel_mgr_(cancel_mgr), in CancellableCall()
42 CancellationToken token = cancel_mgr_->get_cancellation_token(); in Start()
43 const bool not_yet_cancelled = cancel_mgr_->RegisterCallback( in Start()
47 cancel_mgr_->DeregisterCallback(token); in Start()
57 CancellationManager* const cancel_mgr_; // Not owned variable
Dcollective_rma_distributed.cc172 &cancel_mgr_, worker_cache_)); in RecvFromPeer()
183 cancel_mgr_.StartCancel(); in StartAbort()
Dcollective_rma_distributed.h47 CancellationManager cancel_mgr_; variable
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_collective_executor_mgr.cc128 &cancel_mgr_, worker_cache_)); in RecvFromPeer()
139 cancel_mgr_.StartCancel(); in StartAbort()
144 CancellationManager cancel_mgr_; member in tensorflow::__anon0f845a970111::CollectiveRemoteAccessDistributed