Rate this Page

Function at::_amp_foreach_non_finite_check_and_unscale_outf#

Function Documentation#

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