Searched refs:host_alloc_attrs (Results 1 – 4 of 4) sorted by relevance
76 AllocatorAttributes host_alloc_attrs; in RunComposedOp() local77 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()
253 AllocatorAttributes host_alloc_attrs; in ComputeAsync() local254 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()
220 AllocatorAttributes host_alloc_attrs; in ViaDMA() local221 host_alloc_attrs.set_gpu_compatible(true); in ViaDMA()222 host_alloc_attrs.set_on_host(true); in ViaDMA()223 Allocator* cpu_allocator = src->GetAllocator(host_alloc_attrs); in ViaDMA()
421 AllocatorAttributes host_alloc_attrs; in Compute() local422 host_alloc_attrs.set_gpu_compatible(true); in Compute()423 host_alloc_attrs.set_on_host(true); in Compute()424 Allocator* cpu_allocator = ctx->device()->GetAllocator(host_alloc_attrs); in Compute()