Searched refs:CudaAtomicMax (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_device_functions.h | 562 __device__ detail::ToTypeIfConvertible<U, T> CudaAtomicMax(T* ptr, U value) { in CudaAtomicMax() function
566 __device__ inline float CudaAtomicMax(float* ptr, float value) { in CudaAtomicMax() function
571 __device__ inline double CudaAtomicMax(double* ptr, double value) { in CudaAtomicMax() function
576 __device__ inline Eigen::half CudaAtomicMax(Eigen::half* ptr, in CudaAtomicMax() function
583 __device__ inline tensorflow::uint64 CudaAtomicMax(tensorflow::uint64* ptr, in CudaAtomicMax() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops.h | 105 CudaAtomicMax(dest, value); in operator()
|
| D | multinomial_op_gpu.cu.cc | 49 CudaAtomicMax(reinterpret_cast<UnsignedOutputType*>(output + maxima_idx), in MultinomialKernel()
|
| D | scatter_functor_gpu.cu.h | 69 __device__ void operator()(T* dest, T src) const { CudaAtomicMax(dest, src); }
|