Searched refs:atomicReduce (Results 1 – 1 of 1) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorReductionCuda.h | 25 __device__ EIGEN_ALWAYS_INLINE void atomicReduce(T* output, T accum, R& reducer) { in atomicReduce() function
84 __device__ inline void atomicReduce(half2* output, half2 accum, R<half>& reducer) { in atomicReduce() function
104 __device__ inline void atomicReduce(float* output, float accum, SumReducer<float>&) { in atomicReduce() function
175 atomicReduce(output, accum, reducer); in FullReductionKernel()
251 atomicReduce(scratch, accum, reducer); in FullReductionKernelHalfFloat()
433 atomicReduce(&(output[row]), reduced_val, reducer);
531 atomicReduce((half2*)loc, val, reducer);
690 atomicReduce(&(output[input_col]), reduced_val, reducer);
|