Searched defs:gpuAtomicMul (Results 1 – 1 of 1) sorted by relevance
364 inline __device__ at::BFloat16 gpuAtomicMul(at::BFloat16 * address, at::BFloat16 val) { in gpuAtomicMul() function371 inline __device__ double gpuAtomicMul(double * address, double val) { in gpuAtomicMul() function379 inline __device__ float gpuAtomicMul (float * address, float val) { in gpuAtomicMul() function