Home
last modified time | relevance | path

Searched refs:GpuDevicePtr (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h151 GpuDevicePtr base;
190 GpuContext* context, GpuDevicePtr va, const GenericMemoryHandle& handle,
197 static void UnmapMemory(GpuContext* context, GpuDevicePtr va, uint64 bytes);
297 const char* symbol_name, GpuDevicePtr* dptr,
309 GpuDevicePtr location, uint8 value,
315 GpuDevicePtr location,
323 GpuDevicePtr location,
331 GpuDevicePtr location,
340 GpuDevicePtr gpu_src, uint64 size);
342 GpuDevicePtr gpu_dst,
[all …]
Dgpu_types.h52 using GpuDevicePtr = hipDeviceptr_t; variable
70 using GpuDevicePtr = CUdeviceptr;
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_virtual_mem_allocator_test.cc30 using ::stream_executor::gpu::GpuDevicePtr;
72 GpuDevicePtr gpu_buf = reinterpret_cast<GpuDevicePtr>(gpu_block) + 2048; in TEST()
Dgpu_virtual_mem_allocator.cc29 using ::stream_executor::gpu::GpuDevicePtr;
136 GpuDevicePtr next_va = vmem_.base + next_alloc_offset_; in Alloc()
Dgpu_virtual_mem_allocator.h101 stream_executor::gpu::GpuDevicePtr va;