Home
last modified time | relevance | path

Searched defs:to_alloc_attr (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dcollective_rma_distributed.cc44 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvBufCall()
111 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcollective_rma_local.cc29 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()
H A Dcollective_test_util.cc61 const AllocatorAttributes& to_alloc_attr, Tensor* to_tensor, in RecvFromPeer()