Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dcancellable_call.h28 class CancellableCall {
30 CancellableCall(CancellationManager* cancel_mgr, const string& remote_worker, in CancellableCall() function
38 virtual ~CancellableCall() { wc_->ReleaseWorker(remote_worker_, wi_); } in ~CancellableCall()
H A Dcancellable_call.cc19 void CancellableCall::Start(const StatusCallback& done) { in Start()
37 void CancellableCall::Cancel() { in Cancel()
H A Dcollective_param_resolver_distributed.cc32 class CompleteGroupCall : public CancellableCall {
38 : CancellableCall(cancel_mgr, remote_worker, wc) { in CompleteGroupCall()
54 class CompleteInstanceCall : public CancellableCall {
61 : CancellableCall(cancel_mgr, remote_worker, wc) { in CompleteInstanceCall()
H A Dcollective_rma_distributed.cc39 class RecvBufCall : public CancellableCall {
48 : CancellableCall(cancel_mgr, peer_task, wc) { in RecvBufCall()