Searched refs:to_alloc_attr (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 44 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvBufCall() argument
111 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument
117 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
171 [this, state, to_device, to_alloc_attr, to_device_ctx, to_tensor, cpu_dev, in RecvFromPeer()
199 to_device, cpu_attr, to_alloc_attr, dst_tensor, in RecvFromPeer()
216 to_alloc_attr, dst_tensor, client_locality, state->server_attributes, in RecvFromPeer()
|
| D | collective_rma_distributed.h | 43 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
48 auto consumer_callback = [to_tensor, to_device_ctx, to_device, to_alloc_attr, in RecvFromPeer()
72 to_alloc_attr, // dst AllocatorAttributes in RecvFromPeer()
|
| D | collective_rma_local.h | 43 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor,
|
| D | collective_test_util.h | 52 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor,
|
| D | collective_test_util.cc | 61 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer() argument
67 to_alloc_attr, to_tensor, client_locality, dev_to_dev_stream_index, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 282 const AllocatorAttributes& to_alloc_attr,
|