Searched defs:NormOneOps (Results 1 – 1 of 1) sorted by relevance
319 struct NormOneOps { struct320 inline C10_DEVICE acc_t reduce(acc_t acc, scalar_t data, int64_t /*idx*/) const { in reduce()324 inline C10_DEVICE acc_t combine(acc_t a, acc_t b) const { in combine()328 inline C10_DEVICE out_t project(acc_t a) const { in project()332 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx()337 inline C10_DEVICE acc_t warp_shfl_down(acc_t acc, int offset) const { in warp_shfl_down()