Searched refs:uint32_count (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 317 size_t uint32_count);
324 uint8 value, size_t uint32_count,
333 size_t uint32_count,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 468 size_t uint32_count) { in SynchronousMemsetUint32() argument
472 tensorflow::wrap::hipMemsetD32(pointer, value, uint32_count), in SynchronousMemsetUint32()
479 size_t uint32_count, GpuStreamHandle stream) { in AsynchronousMemsetUint8() argument
482 tensorflow::wrap::hipMemsetAsync(location, value, uint32_count, stream), in AsynchronousMemsetUint8()
489 size_t uint32_count, GpuStreamHandle stream) { in AsynchronousMemsetUint32() argument
493 tensorflow::wrap::hipMemsetD32Async(pointer, value, uint32_count, stream), in AsynchronousMemsetUint32()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 634 size_t uint32_count) { in SynchronousMemsetUint32() argument
636 RETURN_IF_CUDA_RES_ERROR(cuMemsetD32(location, value, uint32_count), in SynchronousMemsetUint32()
642 GpuContext* context, CUdeviceptr location, uint8 value, size_t uint32_count, in AsynchronousMemsetUint8() argument
646 cuMemsetD8Async(location, value, uint32_count, stream), in AsynchronousMemsetUint8()
653 size_t uint32_count, CUstream stream) { in AsynchronousMemsetUint32() argument
656 cuMemsetD32Async(location, value, uint32_count, stream), in AsynchronousMemsetUint32()
|