Home
last modified time | relevance | path

Searched refs:host_alloc_attrs (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc252 AllocatorAttributes host_alloc_attrs; in ViaDMA() local
253 host_alloc_attrs.set_gpu_compatible(true); in ViaDMA()
254 host_alloc_attrs.set_on_host(true); in ViaDMA()
255 Allocator* cpu_allocator = src->GetAllocator(host_alloc_attrs); in ViaDMA()
/external/tensorflow/tensorflow/core/kernels/
Dstack_ops.cc291 AllocatorAttributes host_alloc_attrs; in ComputeAsync() local
292 host_alloc_attrs.set_gpu_compatible(true); in ComputeAsync()
293 host_alloc_attrs.set_on_host(true); in ComputeAsync()
294 Allocator* cpu_allocator = device->GetAllocator(host_alloc_attrs); in ComputeAsync()
/external/tensorflow/tensorflow/contrib/verbs/
Drdma.cc1146 AllocatorAttributes host_alloc_attrs; in Clone() local
1147 host_alloc_attrs.set_nic_compatible(true); in Clone()
1148 host_alloc_attrs.set_on_host(true); in Clone()
1149 Allocator* allocator = src_dev_->GetAllocator(host_alloc_attrs); in Clone()