Searched refs:cpu_attr (Results 1 – 2 of 2) sorted by relevance
129 AllocatorAttributes cpu_attr; in RecvFromPeer() local130 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()
590 AllocatorAttributes cpu_attr; in RecvBufAsync() local591 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()