Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DSumKernel.cpp185 struct NanSumLoadPolicy { struct
186 static constexpr int64_t memsize() { in memsize()
190 static scalar_t load(const char * C10_RESTRICT data, int64_t stride, int64_t index) { in load()
197 struct NanSumLoadPolicy<Vectorized<scalar_t>> { struct
198 using vec_t = Vectorized<scalar_t>;
200 static constexpr int64_t memsize() { in memsize()
204 static vec_t load(const char * C10_RESTRICT data, int64_t stride, int64_t index) { in load()