Searched refs:HostRegister (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 134 static bool HostRegister(GpuContext* context, void* location, uint64 bytes);
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 448 return GpuDriver::HostRegister(context_, location, size); in HostMemoryRegister()
|
| D | rocm_driver.cc | 722 /* static */ bool GpuDriver::HostRegister(GpuContext* context, void* location, in HostRegister() function in stream_executor::gpu::GpuDriver
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 567 return GpuDriver::HostRegister(context_, location, size); in HostMemoryRegister()
|
| D | cuda_driver.cc | 913 /* static */ bool GpuDriver::HostRegister(GpuContext* context, void* location, in HostRegister() function in stream_executor::gpu::GpuDriver
|