Searched refs:CancellableCall (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cancellable_call.h | 28 class CancellableCall {
30 CancellableCall(CancellationManager* cancel_mgr, const string& remote_worker, in CancellableCall() function
37 virtual ~CancellableCall() { wc_->ReleaseWorker(remote_worker_, wi_); } in ~CancellableCall()
|
| D | collective_rma_distributed.cc | 33 class RecvBufCall : public CancellableCall {
42 : CancellableCall(cancel_mgr, peer_task, wc) { in RecvBufCall()
|
| D | collective_param_resolver_distributed.cc | 25 class CompleteGroupCall : public CancellableCall {
30 : CancellableCall(cancel_mgr, remote_worker, wc) { in CompleteGroupCall()
46 class CompleteInstanceCall : public CancellableCall {
53 : CancellableCall(cancel_mgr, remote_worker, wc) { in CompleteInstanceCall()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 34 class RecvBufCall : public CancellableCall {
43 : CancellableCall(cancel_mgr, peer_task, wc) { in RecvBufCall()
|