calc_gcd Class — pytorch Architecture

Source Code

aten/src/ATen/native/Math.h lines 1245–1256

template <typename T>
inline typename std::enable_if_t<std::is_integral_v<T>, T>
calc_gcd(T a, T b) {
  a = abs_impl(a);
  b = abs_impl(b);
  while (a != 0) {
    T c = a;
    a = b % a;
    b = c;
  }
  return b;
}