Searched refs:CudaAtomicSub (Results 1 – 3 of 3) sorted by relevance
536 __device__ detail::ToTypeIfConvertible<U, T> CudaAtomicSub(T* ptr, U value) { in CudaAtomicSub() function541 __device__ inline float CudaAtomicSub(float* ptr, float value) { in CudaAtomicSub() function545 __device__ inline double CudaAtomicSub(double* ptr, double value) { in CudaAtomicSub() function549 __device__ inline tensorflow::uint64 CudaAtomicSub(tensorflow::uint64* ptr, in CudaAtomicSub() function554 __device__ inline Eigen::half CudaAtomicSub(Eigen::half* ptr, in CudaAtomicSub() function
54 CudaAtomicSub(out, val); in operator ()()
49 __device__ void operator()(T* dest, T src) const { CudaAtomicSub(dest, src); }