Searched defs:AbsMaxOps (Results 1 – 1 of 1) sorted by relevance
226 struct AbsMaxOps { struct227 inline C10_DEVICE acc_t reduce(acc_t acc, scalar_t data, int64_t /*idx*/) const { in reduce()231 inline C10_DEVICE acc_t combine(acc_t a, acc_t b) const { in combine()235 inline C10_DEVICE out_t project(acc_t a) const { in project()239 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx()244 inline C10_DEVICE acc_t warp_shfl_down(acc_t acc, int offset) const { in warp_shfl_down()