numel Class — pytorch Architecture

Source Code

aten/src/ATen/native/cpu/SumKernel.cpp lines 293–301

template <typename StorePolicy, typename scalar_t, size_t numel>
void store(char * C10_RESTRICT data, int64_t stride, int64_t index,
                  const std::array<scalar_t, numel> &values) {
  auto *base_ptr = data + stride * index;
  for (const auto k : c10::irange(numel)) {
    auto val = values[k];
    StorePolicy::store(base_ptr, stride, k, val);
  }
}