Searched refs:from_alloc_attr (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 74 const AllocatorAttributes& from_alloc_attr, const Tensor* from_tensor, in PostToPeer() argument
79 from_alloc_attr, done); in PostToPeer()
|
| D | base_collective_executor.h | 135 const AllocatorAttributes& from_alloc_attr, in PostToPeer() argument
140 from_device_ctx, from_alloc_attr, from_tensor, in PostToPeer()
|
| D | collective_rma_local.h | 50 const AllocatorAttributes& from_alloc_attr,
|
| D | test_collective_executor_mgr.h | 44 const AllocatorAttributes& from_alloc_attr, in PostToPeer() argument
|
| D | ring_gatherer_test.cc | 83 const AllocatorAttributes& from_alloc_attr, in PostToPeer() argument
90 from_alloc_attr, from_tensor, client_locality, done); in PostToPeer()
|
| D | ring_reducer_test.cc | 83 const AllocatorAttributes& from_alloc_attr, in PostToPeer() argument
90 from_alloc_attr, from_tensor, client_locality, done); in PostToPeer()
|
| D | hierarchical_tree_broadcaster_test.cc | 177 const AllocatorAttributes& from_alloc_attr, in PostToPeer() argument
184 from_alloc_attr, from_tensor, client_locality, done); in PostToPeer()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 247 const AllocatorAttributes& from_alloc_attr,
|