Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dcuda_device_functions.h592 __device__ detail::ToTypeIfConvertible<U, T> CudaAtomicMin(T* ptr, U value) { in CudaAtomicMin() function
596 __device__ inline float CudaAtomicMin(float* ptr, float value) { in CudaAtomicMin() function
601 __device__ inline double CudaAtomicMin(double* ptr, double value) { in CudaAtomicMin() function
606 __device__ inline Eigen::half CudaAtomicMin(Eigen::half* ptr, in CudaAtomicMin() function
613 __device__ inline tensorflow::uint64 CudaAtomicMin(tensorflow::uint64* ptr, in CudaAtomicMin() function
/external/tensorflow/tensorflow/core/kernels/
Dsegment_reduction_ops.h113 CudaAtomicMin(dest, value); in operator()
Dscatter_functor_gpu.cu.h64 __device__ void operator()(T* dest, T src) const { CudaAtomicMin(dest, src); }