pytorch Architecture Docs
Home / Class/ axpy Class — pytorch Architecture

axpy Class — pytorch Architecture

Architecture documentation for the axpy class in CPUBlas.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/native/CPUBlas.h lines 164–174

template<typename scalar_t>
void axpy(int64_t n, scalar_t a, const scalar_t *x, int64_t incx, scalar_t *y, int64_t incy){
  if(n == 1)
  {
    incx = 1;
    incy = 1;
  }
  axpy_stub(
      kCPU, c10::CppTypeToScalarType<scalar_t>::value,
      n, a, x, incx, y, incy);
}

Source

Analyze Your Own Codebase

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

Try Supermodel Free