| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.cc | 28 CancellationManager* cancel_mgr, in CompleteGroupCall() argument
30 : CancellableCall(cancel_mgr, remote_worker, wc) { in CompleteGroupCall()
51 bool is_source, CancellationManager* cancel_mgr, in CompleteInstanceCall() argument
53 : CancellableCall(cancel_mgr, remote_worker, wc) { in CompleteInstanceCall()
96 const string& device, CollectiveParams* cp, CancellationManager* cancel_mgr, in CompleteParamsAsync() argument
100 CompleteGroupDistributed(device, cp, cancel_mgr, in CompleteParamsAsync()
101 [this, device, cp, cancel_mgr, done]( in CompleteParamsAsync()
105 cancel_mgr, done); in CompleteParamsAsync()
114 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteGroupAsync() argument
123 cp.instance.device_names[0], &cp, cancel_mgr, in CompleteGroupAsync()
[all …]
|
| D | collective_param_resolver_distributed.h | 35 CancellationManager* cancel_mgr,
40 CancellationManager* cancel_mgr,
45 CancellationManager* cancel_mgr,
63 CancellationManager* cancel_mgr,
81 CancellationManager* cancel_mgr,
|
| D | cancellable_call.h | 30 CancellableCall(CancellationManager* cancel_mgr, const string& remote_worker, in CancellableCall() argument
32 : cancel_mgr_(cancel_mgr), in CancellableCall()
|
| D | collective_rma_distributed.cc | 41 CancellationManager* cancel_mgr, WorkerCacheInterface* wc) in RecvBufCall() argument
42 : CancellableCall(cancel_mgr, peer_task, wc) { in RecvBufCall()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 160 CancellationManager* cancel_mgr,
167 CancellationManager* cancel_mgr,
174 CancellationManager* cancel_mgr,
270 CancellationManager* cancel_mgr, in CompleteParamsAsync() argument
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local.h | 48 CancellationManager* cancel_mgr,
53 CancellationManager* cancel_mgr,
58 CancellationManager* cancel_mgr,
|
| D | base_collective_executor.cc | 277 const string& device, CollectiveParams* cp, CancellationManager* cancel_mgr, in CompleteParamsAsync() argument
280 cem_->GetParamResolver()->CompleteParamsAsync(device, cp, cancel_mgr, done); in CompleteParamsAsync()
|
| D | base_collective_executor.h | 114 CancellationManager* cancel_mgr,
|
| D | collective_param_resolver_local.cc | 50 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteGroupAsync() argument
585 const string& device, CollectiveParams* cp, CancellationManager* cancel_mgr, in CompleteParamsAsync() argument
602 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteInstanceAsync() argument
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 42 CancellationManager* cancel_mgr, WorkerCacheInterface* wc) in RecvBufCall() argument
43 : CancellableCall(cancel_mgr, peer_task, wc) { in RecvBufCall()
|