vrsqrt Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vml.h lines 44–55

template <typename scalar_t>
inline void vrsqrt(scalar_t* out, scalar_t* in, int64_t size) {
  parallel_for(0, size, 2048, [out, in](int64_t begin, int64_t end) {
    map(
        [](const Vectorized<scalar_t>& x) {
          return Vectorized<scalar_t>((scalar_t)1) / x.sqrt();
        },
        out + begin,
        in + begin,
        end - begin);
  });
}