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)#