check_both_inf Class — pytorch Architecture

Source Code

aten/src/ATen/test/vec_test_all_types.h lines 294–300

template <typename T>
bool check_both_inf(T x, T y) {
    if constexpr (std::is_floating_point_v<T>) {
        return std::isinf(x) && std::isinf(y);
    }
    return false;
}