Searched refs:FullReductionKernel (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorReductionCuda.h | 125 __global__ void FullReductionKernel(Reducer reducer, const Self input, Index num_coeffs, in FullReductionKernel() function
300 LAUNCH_CUDA_KERNEL((FullReductionKernel<block_size, num_per_thread, Self, Op, Index>),
|
| D | TensorReduction.h | 326 __global__ void FullReductionKernel(R, const S, I, typename S::CoeffReturnType*, unsigned int*);
683 …, int N, typename S, typename R, typename I> friend void internal::FullReductionKernel(R, const S,…
|