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)