Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsegment_reduction_ops.h97 CudaAtomicMul(dest, value); in operator()
Dscatter_functor_gpu.cu.h54 __device__ void operator()(T* dest, T src) const { CudaAtomicMul(dest, src); }
/external/tensorflow/tensorflow/core/util/
Dcuda_device_functions.h622 __device__ detail::ToTypeIfConvertible<U, T> CudaAtomicMul(T* ptr, U value) { in CudaAtomicMul() function