Change customize_loss_grad to use_default_grad_scale.
#10223
Merged
customize_loss_grad to use_default_grad_scale.
#10223