Searched defs:to_alloc_attr (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | collective_rma_distributed.cc | 38 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvBufCall() 78 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
|
/external/tensorflow/tensorflow/contrib/gdr/ |
D | gdr_collective_executor_mgr.cc | 39 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 | base_collective_executor.h | 124 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
|
D | collective_rma_local.cc | 29 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
|
D | test_collective_executor_mgr.h | 34 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
|
D | ring_gatherer_test.cc | 69 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
|
D | ring_reducer_test.cc | 69 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
|
D | hierarchical_tree_broadcaster_test.cc | 165 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
|