Home
last modified time | relevance | path

Searched refs:host_mem (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_virtual_mem_allocator_test.cc67 void* host_mem[2] = {GpuDriver::HostAllocate(gpu_context, kBufSize), in TEST() local
69 std::memset(host_mem[0], 'z', kBufSize); in TEST()
70 std::memset(host_mem[1], 0, kBufSize); in TEST()
73 ASSERT_TRUE(GpuDriver::SynchronousMemcpyH2D(gpu_context, gpu_buf, host_mem[0], in TEST()
76 ASSERT_TRUE(GpuDriver::SynchronousMemcpyD2H(gpu_context, host_mem[1], gpu_buf, in TEST()
80 ASSERT_EQ('z', reinterpret_cast<const char*>(host_mem[1])[i]); in TEST()
/external/crosvm/rutabaga_gfx/src/
Drutabaga_2d.rs167 host_mem: vec![0; resource_size], in create_3d()
225 VolatileSlice::new(info_2d.host_mem.as_mut_slice()), in transfer_write()
265 [VolatileSlice::new(info_2d.host_mem.as_mut_slice())] in transfer_read()
Drutabaga_core.rs28 pub host_mem: Vec<u8>, field
/external/linux-kselftest/tools/testing/selftests/kvm/lib/
Dkvm_util.c657 region->host_mem = align(region->mmap_start, alignment); in vm_userspace_mem_region_add()
661 ret = madvise(region->host_mem, npages * vm->page_size, in vm_userspace_mem_region_add()
667 region->host_mem, npages * vm->page_size, src_type); in vm_userspace_mem_region_add()
677 region->region.userspace_addr = (uintptr_t) region->host_mem; in vm_userspace_mem_region_add()
1036 return (void *) ((uintptr_t) region->host_mem in addr_gpa2hva()
1066 if ((hva >= region->host_mem) in addr_hva2gpa()
1067 && (hva <= (region->host_mem in addr_hva2gpa()
1071 + (hva - (uintptr_t) region->host_mem)); in addr_hva2gpa()
1463 region->host_mem); in vm_dump()
Dkvm_util_internal.h32 void *host_mem; member
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc649 void* host_mem = nullptr; in HostAllocate() local
652 tensorflow::wrap::hipHostMalloc(&host_mem, bytes, hipHostMallocPortable); in HostAllocate()
657 return host_mem; in HostAllocate()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc847 void* host_mem = nullptr; in HostAllocate() local
849 CUresult res = cuMemHostAlloc(&host_mem, bytes, CU_MEMHOSTALLOC_PORTABLE); in HostAllocate()
854 return host_mem; in HostAllocate()