Searched defs:InnerSumCastLoadPolicy (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | SumKernel.cpp | 89 struct InnerSumCastLoadPolicy <vec_t, vacc_t, struct
92 using scalar_t = vechold_type<vec_t>;
93 using acc_t = vechold_type<vacc_t>;
95 static constexpr int64_t memsize() { in memsize()
99 static vacc_t load(const char * C10_RESTRICT data, int64_t stride, int64_t index) { in load()
108 struct InnerSumCastLoadPolicy<scalar_t, scalar_t, void>: struct
113 struct InnerSumCastLoadPolicy <vec_t, vacc_t, std::enable_if_t<is_reduced_floating_point_v<vechold_… struct
114 using scalar_t = vechold_type<vec_t>;
116 static constexpr int64_t memsize() { in memsize()
120 static vacc_t load(const char * C10_RESTRICT data, int64_t stride, int64_t index) { in load()
|