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()
555 context_, AsCudaDevicePtr(location), 0x0, size / 4); in SynchronousMemZero()
557 return GpuDriver::SynchronousMemsetUint8(context_, AsCudaDevicePtr(location), in SynchronousMemZero()
570 context_, AsCudaDevicePtr(location), pattern, size / 4); in SynchronousMemSet()
572 return GpuDriver::SynchronousMemsetUint8(context_, AsCudaDevicePtr(location), in SynchronousMemSet()
578 return GpuDriver::SynchronousMemcpyH2D(context_, AsCudaDevicePtr(gpu_dst), in SynchronousMemcpy()
586 AsCudaDevicePtr(gpu_src), size); in SynchronousMemcpy()
591 return GpuDriver::SynchronousMemcpyD2D(context_, AsCudaDevicePtr(gpu_dst), in SynchronousMemcpyDeviceToDevice()
[all …]
|