Function at::_amp_foreach_non_finite_check_and_unscale_out

• Defined in File Functions.h

Function Documentation

inline void at::_amp_foreach_non_finite_check_and_unscale_out(at::TensorList out, at::TensorList self, at::Tensor &found_inf, const at::Tensor &inv_scale)