Home
last modified time | relevance | path

Searched refs:prod_attr (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dbuf_rendezvous.h63 AllocatorAttributes prod_attr; member
Dcollective_rma_local.cc54 hook->prod_attr, // src AllocatorAttributes in RecvFromPeer()
Dbuf_rendezvous.cc93 h->prod_attr = attr; in ProvideBuf()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_worker.cc177 hook->prod_dev, hook->prod_ctx, hook->prod_attr.on_host(), in RecvBufAsync()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_service.cc585 hook->prod_attr.on_host(); in RecvBufAsync()