Searched defs:NormTwoOps (Results 1 – 1 of 1) sorted by relevance
367 struct NormTwoOps { struct368 inline C10_DEVICE acc_t reduce(acc_t acc, scalar_t data, int64_t /*idx*/) const { in reduce()373 inline C10_DEVICE acc_t combine(acc_t a, acc_t b) const { in combine()377 inline C10_DEVICE out_t project(acc_t a) const { in project()381 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx()386 inline C10_DEVICE acc_t warp_shfl_down(acc_t acc, int offset) const { in warp_shfl_down()