Searched defs:NormZeroOps (Results 1 – 1 of 1) sorted by relevance
289 struct NormZeroOps { struct290 inline C10_DEVICE acc_t reduce(acc_t acc, scalar_t data, int64_t /*idx*/) const { in reduce()294 inline C10_DEVICE acc_t combine(acc_t a, acc_t b) const { in combine()298 inline C10_DEVICE out_t project(acc_t a) const { in project()302 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx()308 inline C10_DEVICE acc_t warp_shfl_down(acc_t acc, int offset) const { in warp_shfl_down()