Home
last modified time | relevance | path

Searched refs:cpu_attr (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_rma_distributed.cc129 AllocatorAttributes cpu_attr; in RecvFromPeer() local
130 cpu_attr.set_gpu_compatible(true); in RecvFromPeer()
131 Tensor* cpu_tensor = new Tensor(cpu_dev->GetAllocator(cpu_attr), in RecvFromPeer()
137 to_device, cpu_attr, to_alloc_attr, cpu_tensor, in RecvFromPeer()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_service.cc590 AllocatorAttributes cpu_attr; in RecvBufAsync() local
591 cpu_attr.set_gpu_compatible(true); in RecvBufAsync()
592 cpu_attr.set_nic_compatible(true); in RecvBufAsync()
593 Tensor* cpu_tensor = new Tensor(cpu_dev->GetAllocator(cpu_attr), in RecvBufAsync()