Home
last modified time | relevance | path

Searched defs:MinMaxOps (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DSharedReduceOps.h511 struct MinMaxOps { struct
512 using acc_t = detail::pair<acc_scalar_t, acc_scalar_t>;
513 inline C10_DEVICE acc_t reduce(acc_t acc, scalar_t data, index_t /*idx*/) const { in reduce()
517 inline C10_DEVICE acc_t combine(acc_t a, acc_t b) const { in combine()
524 inline C10_DEVICE acc_t project(acc_t acc) const { in project()
528 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx()
533 inline C10_DEVICE acc_t warp_shfl_down(acc_t acc, int offset) const { in warp_shfl_down()