Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_virtual_mem_allocator.h79 stream_executor::gpu::GpuDriver::VmemSpan vmem, size_t granularity);
91 stream_executor::gpu::GpuDriver::VmemSpan vmem_;
Dgpu_virtual_mem_allocator.cc100 GpuDriver::VmemSpan vmem, in Create()
115 GpuDriver::VmemSpan vmem, size_t granularity) in GpuVirtualMemAllocator()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h150 struct VmemSpan { struct
155 static port::StatusOr<VmemSpan> ReserveVirtualMemory(GpuContext* context, argument
161 static void FreeVirtualMemory(GpuContext* context, VmemSpan reservation);
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc894 /* static */ port::StatusOr<GpuDriver::VmemSpan>
909 GpuContext* context, GpuDriver::VmemSpan reservation) { in FreeVirtualMemory()