Searched defs:InnerNanSumCastLoadPolicy (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | SumKernel.cpp | 226 struct InnerNanSumCastLoadPolicy <vec_t, vacc_t, struct
229 using scalar_t = vechold_type<vec_t>;
230 using acc_t = vechold_type<vacc_t>;
232 static constexpr int64_t memsize() { in memsize()
236 static vacc_t load(const char * C10_RESTRICT data, int64_t stride, int64_t index) { in load()
245 struct InnerNanSumCastLoadPolicy<scalar_t, scalar_t, void>: struct
250 struct InnerNanSumCastLoadPolicy <vec_t, vacc_t, std::enable_if_t<is_reduced_floating_point_v<vecho… struct
251 using scalar_t = vechold_type<vec_t>;
253 static constexpr int64_t memsize() { in memsize()
257 static vacc_t load(const char * C10_RESTRICT data, int64_t stride, int64_t index) { in load()
|