Home / Class/ vec_fun Class — pytorch Architecture

vec_fun Class — pytorch Architecture

Architecture documentation for the vec_fun class in functional_base.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/cpu/vec/functional_base.h lines 46–53

template <typename scalar_t, typename Op>
struct VecReduceAllSIMD {
  static inline scalar_t apply(
      const Op& vec_fun,
      const Vectorized<scalar_t>& acc_vec) {
    return vec_reduce_all(vec_fun, acc_vec, Vectorized<scalar_t>::size());
  }
};

Source

Analyze Your Own Codebase

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

Try Supermodel Free