| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 40 const DeviceLocality& client_locality, in RecvBufCall() argument
46 *req_.mutable_client_locality() = client_locality; in RecvBufCall()
82 const DeviceLocality& client_locality, in RecvFromPeer() argument
88 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
120 to_tensor, client_locality, in RecvFromPeer()
127 to_alloc_attr, to_tensor, client_locality, state->server_locality, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 39 const DeviceLocality& client_locality, in RecvBufCall() argument
45 *req_.mutable_client_locality() = client_locality; in RecvBufCall()
79 const DeviceLocality& client_locality, int dev_to_dev_stream_index, in RecvFromPeer() argument
84 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
164 to_tensor, client_locality, in RecvFromPeer()
171 to_alloc_attr, to_tensor, client_locality, state->server_locality, in RecvFromPeer()
|
| D | collective_rma_distributed.h | 39 const DeviceLocality& client_locality,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.h | 125 const DeviceLocality& client_locality, int stream_index, in RecvFromPeer() argument
129 to_alloc_attr, to_tensor, client_locality, stream_index, done); in RecvFromPeer()
137 const DeviceLocality& client_locality, in PostToPeer() argument
141 client_locality, done); in PostToPeer()
|
| D | collective_rma_local.h | 43 const DeviceLocality& client_locality,
52 const DeviceLocality& client_locality,
|
| D | test_collective_executor_mgr.h | 35 const DeviceLocality& client_locality, in RecvFromPeer() argument
46 const DeviceLocality& client_locality, in PostToPeer() argument
|
| D | collective_rma_local.cc | 30 const DeviceLocality& client_locality, int dev_to_dev_stream_index, in RecvFromPeer() argument
75 const DeviceLocality& client_locality, const StatusCallback& done) { in PostToPeer() argument
|
| D | ring_gatherer_test.cc | 70 const DeviceLocality& client_locality, in RecvFromPeer() argument
76 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
85 const DeviceLocality& client_locality, in PostToPeer() argument
90 from_alloc_attr, from_tensor, client_locality, done); in PostToPeer()
|
| D | ring_reducer_test.cc | 70 const DeviceLocality& client_locality, in RecvFromPeer() argument
76 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
85 const DeviceLocality& client_locality, in PostToPeer() argument
90 from_alloc_attr, from_tensor, client_locality, done); in PostToPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 166 const DeviceLocality& client_locality, int stream_index, in RecvFromPeer() argument
171 to_alloc_attr, to_tensor, client_locality, stream_index, done); in RecvFromPeer()
179 const DeviceLocality& client_locality, in PostToPeer() argument
184 from_alloc_attr, from_tensor, client_locality, done); in PostToPeer()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 240 const DeviceLocality& client_locality,
249 const DeviceLocality& client_locality,
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 325 DeviceLocality client_locality = 4; field
459 DeviceLocality client_locality = 5; field
|