| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 39 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvBufCall() argument
81 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument
88 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
102 auto recv_buf_callback = [this, state, peer_task, to_device, to_alloc_attr, in RecvFromPeer()
107 to_device_ctx, to_alloc_attr.on_host(), done); in RecvFromPeer()
119 to_device, to_device_ctx, to_alloc_attr, 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 | 38 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvBufCall() argument
78 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument
84 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
98 auto recv_buf_callback = [this, state, peer_task, to_device, to_alloc_attr, in RecvFromPeer()
137 to_device, cpu_attr, to_alloc_attr, cpu_tensor, in RecvFromPeer()
163 to_device, to_device_ctx, to_alloc_attr, in RecvFromPeer()
171 to_alloc_attr, to_tensor, client_locality, state->server_locality, in RecvFromPeer()
|
| D | collective_rma_distributed.h | 38 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 29 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument
41 key, [to_tensor, to_device_ctx, to_device, to_alloc_attr, in RecvFromPeer()
55 to_alloc_attr, // dst AllocatorAttributes in RecvFromPeer()
|
| D | base_collective_executor.h | 124 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument
129 to_alloc_attr, to_tensor, client_locality, stream_index, done); in RecvFromPeer()
|
| D | collective_rma_local.h | 42 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor,
|
| D | test_collective_executor_mgr.h | 34 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument
|
| D | ring_gatherer_test.cc | 69 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument
76 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
|
| D | ring_reducer_test.cc | 69 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument
76 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 165 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument
171 to_alloc_attr, to_tensor, client_locality, stream_index, done); in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 238 const AllocatorAttributes& to_alloc_attr,
|