| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 35 RecvBufCall(int64 step_id, const string& peer_device, const string& peer_task, in RecvBufCall() argument
42 : CancellableCall(cancel_mgr, peer_task, wc) { in RecvBufCall()
77 const string& peer_device, const string& peer_task, bool peer_is_local, in RecvFromPeer() argument
84 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
99 auto recv_buf_callback = [this, state, peer_task, to_device, to_alloc_attr, in RecvFromPeer()
154 dev_resolver_->ClearTask(peer_task); in RecvFromPeer()
163 auto dev_attributes_callback = [this, state, peer_device, peer_task, key, in RecvFromPeer()
171 step_id_, peer_device, peer_task, key, to_device, to_device_ctx, in RecvFromPeer()
178 dev_resolver_->GetDeviceAttributesAsync(peer_device, peer_task, in RecvFromPeer()
|
| D | collective_rma_distributed.h | 37 void RecvFromPeer(const string& peer_device, const string& peer_task,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.h | 122 void RecvFromPeer(const string& peer_device, const string& peer_task, in RecvFromPeer() argument
129 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
133 void PostToPeer(const string& peer_device, const string& peer_task, in PostToPeer() argument
140 remote_access_->PostToPeer(peer_device, peer_task, key, from_device, in PostToPeer()
|
| D | collective_rma_local.h | 43 void RecvFromPeer(const string& peer_device, const string& peer_task,
51 void PostToPeer(const string& peer_device, const string& peer_task,
|
| D | test_collective_executor_mgr.h | 31 void RecvFromPeer(const string& peer_device, const string& peer_task, in RecvFromPeer() argument
41 void PostToPeer(const string& peer_device, const string& peer_task, in PostToPeer() argument
|
| D | collective_rma_local.cc | 30 const string& peer_device, const string& peer_task, bool peer_is_local, in RecvFromPeer() argument
101 const string& peer_device, const string& peer_task, const string& key, in PostToPeer() argument
|
| D | ring_gatherer_test.cc | 69 void RecvFromPeer(const string& peer_device, const string& peer_task, in RecvFromPeer() argument
78 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
83 void PostToPeer(const string& peer_device, const string& peer_task, in PostToPeer() argument
92 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| D | ring_reducer_test.cc | 69 void RecvFromPeer(const string& peer_device, const string& peer_task, in RecvFromPeer() argument
78 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
83 void PostToPeer(const string& peer_device, const string& peer_task, in PostToPeer() argument
92 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 165 void RecvFromPeer(const string& peer_device, const string& peer_task, in RecvFromPeer() argument
173 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
177 void PostToPeer(const string& peer_device, const string& peer_task, in PostToPeer() argument
186 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 248 virtual void RecvFromPeer(const string& peer_device, const string& peer_task,
257 virtual void PostToPeer(const string& peer_device, const string& peer_task,
|