Home
last modified time | relevance | path

Searched refs:GpuAtomicCasHelper (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dgpu_device_functions.h585 __device__ T GpuAtomicCasHelper(T* ptr, F accumulate) { in GpuAtomicCasHelper() function
594 CREATE_CUDA_DEVICE_FUNCTION_ALIAS(GpuAtomicCasHelper, CudaAtomicCasHelper);
599 __device__ float GpuAtomicCasHelper(float* ptr, F accumulate) { in GpuAtomicCasHelper() function
601 GpuAtomicCasHelper(reinterpret_cast<int32*>(ptr), [accumulate](int32 a) { in GpuAtomicCasHelper()
606 __device__ double GpuAtomicCasHelper(double* ptr, F accumulate) { in GpuAtomicCasHelper() function
612 GpuAtomicCasHelper(reinterpret_cast<unsigned long long*>(ptr), in GpuAtomicCasHelper()
619 return __longlong_as_double(GpuAtomicCasHelper( in GpuAtomicCasHelper()
640 __device__ Eigen::half GpuAtomicCasHelper(Eigen::half* ptr, F accumulate) { in GpuAtomicCasHelper() function
650 uint32 result = GpuAtomicCasHelper(address, [accumulate](uint32 arg) { in GpuAtomicCasHelper()
659 uint32 result = GpuAtomicCasHelper(address, [accumulate](uint32 arg) { in GpuAtomicCasHelper()
[all …]