Home / Class/ init Class — pytorch Architecture

init Class — pytorch Architecture

Architecture documentation for the init class in ReduceUtils.h from the pytorch codebase.

Entity Profile

Source Code

aten/src/ATen/native/cpu/ReduceUtils.h lines 71–79

template <typename scalar_t>
inline void init(scalar_t* out, int64_t size, const vec_scalar_t<scalar_t>& val) {
  using Vec = Vectorized<vec_scalar_t<scalar_t>>;
  map<scalar_t>(
      [val](Vec x) { return Vec(val); },
      out,
      out,
      size);
}

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free