Searched defs:scaled_grad (Results 1 – 2 of 2) sorted by relevance
43 void _amp_non_finite_check_and_unscale_cuda_(Tensor& scaled_grad, in _amp_non_finite_check_and_unscale_cuda_()
2284 auto scaled_grad = in matchAndRewrite() local