Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h173 struct GenericMemoryHandle { struct
177 static port::StatusOr<GenericMemoryHandle> CreateMemoryHandle( argument
183 GenericMemoryHandle handle);
190 GpuContext* context, GpuDevicePtr va, const GenericMemoryHandle& handle,
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_virtual_mem_allocator.h102 stream_executor::gpu::GpuDriver::GenericMemoryHandle physical;
Dgpu_virtual_mem_allocator.cc156 GpuDriver::GenericMemoryHandle handle = std::move(maybe_handle).ValueOrDie(); in Alloc()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc935 /* static */ port::StatusOr<GpuDriver::GenericMemoryHandle>
956 return GpuDriver::GenericMemoryHandle{mem_handle, bytes}; in CreateMemoryHandle()
960 GpuContext* context, GpuDriver::GenericMemoryHandle handle) { in ReleaseMemoryHandle()
972 const GpuDriver::GenericMemoryHandle& handle, in MapMemory()