Searched refs:gpu_context_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_virtual_mem_allocator.cc | 117 gpu_context_(gpu_context), in GpuVirtualMemAllocator()
125 GpuDriver::UnmapMemory(&gpu_context_, mapping.va, mapping.physical.bytes); in ~GpuVirtualMemAllocator()
126 GpuDriver::ReleaseMemoryHandle(&gpu_context_, std::move(mapping.physical)); in ~GpuVirtualMemAllocator()
128 GpuDriver::FreeVirtualMemory(&gpu_context_, vmem_); in ~GpuVirtualMemAllocator()
151 GpuDriver::CreateMemoryHandle(&gpu_context_, padded_bytes); in Alloc()
160 GpuDriver::MapMemory(&gpu_context_, next_va, handle, access_gpu_handles_); in Alloc()
163 GpuDriver::ReleaseMemoryHandle(&gpu_context_, std::move(handle)); in Alloc()
205 GpuDriver::UnmapMemory(&gpu_context_, it->va, it->physical.bytes); in Free()
206 GpuDriver::ReleaseMemoryHandle(&gpu_context_, std::move(it->physical)); in Free()
|
| D | gpu_virtual_mem_allocator.h | 81 stream_executor::gpu::GpuContext& gpu_context_; variable
|