Searched defs:DistReduceOp (Results 1 – 1 of 1) sorted by relevance
84 struct DistReduceOp { struct85 __forceinline__ __device__ scalar_t combine(scalar_t a, scalar_t b) const { in combine()90 __forceinline__ __device__ scalar_t warp_shfl_down(scalar_t data, int offset) const { in warp_shfl_down()