chbevl Class — pytorch Architecture

Source Code

aten/src/ATen/native/Math.h lines 1294–1309

template <typename T>
inline typename std::enable_if_t<std::is_floating_point_v<T>, T>
chbevl(const T x, const T array[], size_t len) {
  T b0, b1, b2;

  b0 = array[0];
  b1 = static_cast<T>(0.0);

  for (size_t i = 1; i < len; ++i) {
    b2 = b1;
    b1 = b0;
    b0 = x * b1 - b2 + array[i];
  }

  return (static_cast<T>(0.5) * (b0 - b2));
}