Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dunary_ops_composition_test.cc76 AllocatorAttributes host_alloc_attrs; in RunComposedOp() local
77 host_alloc_attrs.set_gpu_compatible(true); in RunComposedOp()
78 host_alloc_attrs.set_on_host(true); in RunComposedOp()
79 Allocator* cpu_allocator = device_->GetAllocator(host_alloc_attrs); in RunComposedOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstack.cc253 AllocatorAttributes host_alloc_attrs; in ComputeAsync() local
254 host_alloc_attrs.set_gpu_compatible(true); in ComputeAsync()
255 host_alloc_attrs.set_on_host(true); in ComputeAsync()
256 Allocator* cpu_allocator = device->GetAllocator(host_alloc_attrs); in ComputeAsync()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcopy_tensor.cc222 AllocatorAttributes host_alloc_attrs; in ViaDMA() local
223 host_alloc_attrs.set_gpu_compatible(true); in ViaDMA()
224 host_alloc_attrs.set_on_host(true); in ViaDMA()
225 Allocator* cpu_allocator = src->GetAllocator(host_alloc_attrs); in ViaDMA()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/kernels/
H A Dxla_ops.cc535 AllocatorAttributes host_alloc_attrs; in Compute() local
536 host_alloc_attrs.set_gpu_compatible(true); in Compute()
537 host_alloc_attrs.set_on_host(true); in Compute()
538 Allocator* cpu_allocator = ctx->device()->GetAllocator(host_alloc_attrs); in Compute()