Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dstack.cc252 AllocatorAttributes host_alloc_attrs; in ComputeAsync() local
253 host_alloc_attrs.set_gpu_compatible(true); in ComputeAsync()
254 host_alloc_attrs.set_on_host(true); in ComputeAsync()
255 Allocator* cpu_allocator = device->GetAllocator(host_alloc_attrs); in ComputeAsync()
/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc267 AllocatorAttributes host_alloc_attrs; in ViaDMA() local
268 host_alloc_attrs.set_gpu_compatible(true); in ViaDMA()
269 host_alloc_attrs.set_on_host(true); in ViaDMA()
270 Allocator* cpu_allocator = src->GetAllocator(host_alloc_attrs); in ViaDMA()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc475 AllocatorAttributes host_alloc_attrs; in Compute() local
476 host_alloc_attrs.set_gpu_compatible(true); in Compute()
477 host_alloc_attrs.set_on_host(true); in Compute()
478 Allocator* cpu_allocator = ctx->device()->GetAllocator(host_alloc_attrs); in Compute()
/external/tensorflow/tensorflow/contrib/verbs/
Drdma.cc1148 AllocatorAttributes host_alloc_attrs; in Clone() local
1149 host_alloc_attrs.set_nic_compatible(true); in Clone()
1150 host_alloc_attrs.set_on_host(true); in Clone()
1151 Allocator* allocator = src_dev_->GetAllocator(host_alloc_attrs); in Clone()