| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 36 RecvBufCall(int64 step_id, const string& peer_device, const string& peer_task, in RecvBufCall() argument
43 : CancellableCall(cancel_mgr, peer_task, wc) { in RecvBufCall()
78 void RecvFromPeer(const string& peer_device, const string& peer_task, in RecvFromPeer() argument
87 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
102 auto recv_buf_callback = [this, state, peer_task, to_device, to_alloc_attr, in RecvFromPeer()
110 dev_resolver_->ClearTask(peer_task); in RecvFromPeer()
118 auto dev_locality_callback = [this, state, peer_device, peer_task, key, in RecvFromPeer()
126 step_id_, peer_device, peer_task, key, to_device, to_device_ctx, in RecvFromPeer()
134 peer_device, peer_task, &state->server_locality, dev_locality_callback); in RecvFromPeer()
|
| /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()
76 const string& peer_device, const string& peer_task, bool peer_is_local, in RecvFromPeer() argument
83 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
98 auto recv_buf_callback = [this, state, peer_task, to_device, to_alloc_attr, in RecvFromPeer()
153 dev_resolver_->ClearTask(peer_task); in RecvFromPeer()
162 auto dev_locality_callback = [this, state, peer_device, peer_task, key, in RecvFromPeer()
170 step_id_, peer_device, peer_task, key, to_device, to_device_ctx, in RecvFromPeer()
178 peer_device, peer_task, &state->server_locality, dev_locality_callback); in RecvFromPeer()
|
| D | collective_rma_distributed.h | 35 void RecvFromPeer(const string& peer_device, const string& peer_task,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.h | 121 void RecvFromPeer(const string& peer_device, const string& peer_task, in RecvFromPeer() argument
128 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
132 void PostToPeer(const string& peer_device, const string& peer_task, in PostToPeer() argument
139 remote_access_->PostToPeer(peer_device, peer_task, key, from_device, in PostToPeer()
|
| D | collective_rma_local.h | 39 void RecvFromPeer(const string& peer_device, const string& peer_task,
47 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 | 27 const string& peer_device, const string& peer_task, bool peer_is_local, in RecvFromPeer() argument
72 const string& peer_device, const string& peer_task, const string& key, in PostToPeer() argument
|
| D | ring_gatherer_test.cc | 66 void RecvFromPeer(const string& peer_device, const string& peer_task, in RecvFromPeer() argument
75 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
80 void PostToPeer(const string& peer_device, const string& peer_task, in PostToPeer() argument
89 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| D | ring_reducer_test.cc | 66 void RecvFromPeer(const string& peer_device, const string& peer_task, in RecvFromPeer() argument
75 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
80 void PostToPeer(const string& peer_device, const string& peer_task, in PostToPeer() argument
89 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 162 void RecvFromPeer(const string& peer_device, const string& peer_task, in RecvFromPeer() argument
170 peer_device, peer_task, peer_is_local, key, to_device, to_device_ctx, in RecvFromPeer()
174 void PostToPeer(const string& peer_device, const string& peer_task, in PostToPeer() argument
183 peer_device, peer_task, key, from_device, from_device_ctx, in PostToPeer()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 235 virtual void RecvFromPeer(const string& peer_device, const string& peer_task,
244 virtual void PostToPeer(const string& peer_device, const string& peer_task,
|