Searched defs:OuterSumCastLoadPolicy (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | SumKernel.cpp | 133 struct OuterSumCastLoadPolicy <vec_t, vacc_t, struct
137 using scalar_t = vechold_type<vec_t>;
138 using acc_t = vechold_type<vacc_t>;
140 static constexpr int64_t memsize() { in memsize()
144 static vacc_t load(const char * C10_RESTRICT data, int64_t stride, int64_t index) { in load()
160 struct OuterSumCastLoadPolicy <vec_t, vacc_t, std::enable_if_t<is_reduced_floating_point_v<vechold_… struct
161 using scalar_t = vechold_type<vec_t>;
163 static constexpr int64_t memsize() { in memsize()
167 static vacc_t load(const char * C10_RESTRICT data, int64_t stride, int64_t index) { in load()
176 struct OuterSumCastLoadPolicy<scalar_t, scalar_t, void>: struct
|