Function at::_amp_foreach_non_finite_check_and_unscale#

• Defined in File Functions.h

Function Documentation#

inline ::std::tuple<::std::vector<at::Tensor>, at::Tensor> at::_amp_foreach_non_finite_check_and_unscale(at::TensorList self, const at::Tensor &found_inf, const at::Tensor &inv_scale)#