Searched defs:NanSumOps (Results 1 – 1 of 1) sorted by relevance
393 struct NanSumOps { struct394 inline C10_DEVICE acc_t reduce(acc_t a, data_t b, int64_t /*idx*/) const { in reduce()398 inline C10_DEVICE acc_t combine(acc_t a, acc_t b) const { in combine()402 inline C10_DEVICE data_t project(acc_t a) const { in project()406 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx()411 inline C10_DEVICE acc_t warp_shfl_down(acc_t data, int offset) const { in warp_shfl_down()