dot_impl Class — pytorch Architecture

Source Code

aten/src/ATen/native/BlasKernel.cpp lines 689–696

template <typename scalar_t>
scalar_t dot_impl(int64_t n, const scalar_t* x, int64_t incx, const scalar_t* y, int64_t incy) {
  if (n == 1) {
    incx = 1;
    incy = 1;
  }
  return blas_impl::dot_naive(n, x, incx, y, incy, std::multiplies<at::opmath_type<scalar_t>>{});
}