Searched refs:uint32_count (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 259 size_t uint32_count);
266 size_t uint32_count,
274 size_t uint32_count,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 514 size_t uint32_count) { in SynchronousMemsetUint32() argument
517 hipError_t res = tensorflow::wrap::hipMemsetD32(pointer, value, uint32_count); in SynchronousMemsetUint32()
528 size_t uint32_count, in AsynchronousMemsetUint8() argument
532 tensorflow::wrap::hipMemsetAsync(location, value, uint32_count, stream); in AsynchronousMemsetUint8()
544 size_t uint32_count, in AsynchronousMemsetUint32() argument
549 tensorflow::wrap::hipMemsetD32Async(pointer, value, uint32_count, stream); in AsynchronousMemsetUint32()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 677 size_t uint32_count) { in SynchronousMemsetUint32() argument
679 CUresult res = cuMemsetD32(location, value, uint32_count); in SynchronousMemsetUint32()
690 size_t uint32_count, in AsynchronousMemsetUint8() argument
693 CUresult res = cuMemsetD8Async(location, value, uint32_count, stream); in AsynchronousMemsetUint8()
705 size_t uint32_count, in AsynchronousMemsetUint32() argument
708 CUresult res = cuMemsetD32Async(location, value, uint32_count, stream); in AsynchronousMemsetUint32()
|