Searched refs:AsCudaDevicePtr (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 104 static CUdeviceptr AsCudaDevicePtr(const DeviceMemoryBase &gpu_mem) { in AsCudaDevicePtr() function
109 static CUdeviceptr AsCudaDevicePtr(DeviceMemoryBase *gpu_mem) { in AsCudaDevicePtr() function
110 return AsCudaDevicePtr(*gpu_mem); in AsCudaDevicePtr()
583 context_, AsCudaDevicePtr(location), 0x0, size / 4); in SynchronousMemZero()
585 return GpuDriver::SynchronousMemsetUint8(context_, AsCudaDevicePtr(location), in SynchronousMemZero()
598 context_, AsCudaDevicePtr(location), pattern, size / 4); in SynchronousMemSet()
600 return GpuDriver::SynchronousMemsetUint8(context_, AsCudaDevicePtr(location), in SynchronousMemSet()
606 return GpuDriver::SynchronousMemcpyH2D(context_, AsCudaDevicePtr(gpu_dst), in SynchronousMemcpy()
614 AsCudaDevicePtr(gpu_src), size); in SynchronousMemcpy()
619 return GpuDriver::SynchronousMemcpyD2D(context_, AsCudaDevicePtr(gpu_dst), in SynchronousMemcpyDeviceToDevice()
[all …]
|