Searched defs:to_alloc_attr (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 40 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvBufCall()
81 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 29 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
|
| D | permuter_test.cc | 78 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
|
| D | ring_gatherer_test.cc | 70 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
|
| D | ring_reducer_test.cc | 71 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 168 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
|