Searched defs:func_wrapper_t (Results 1 – 1 of 1) sorted by relevance
251 struct func_wrapper_t { struct252 using arg_t = typename binary_function_traits<func_t>::arg1_t;253 using scalar_t = typename binary_function_traits<func_t>::arg2_t;255 func_t combine;256 static inline __device__ out_scalar_t project(arg_t arg) { in project()259 static inline __device__ arg_t warp_shfl_down(arg_t arg, int offset) { in warp_shfl_down()263 static __device__ arg_t translate_idx(arg_t acc, int64_t /*idx*/) { in translate_idx()267 func_wrapper_t(const func_t& op) : combine(op) { in func_wrapper_t() function271 __device__ arg_t reduce(arg_t acc, scalar_t val, int64_t idx) const { in reduce()