      # Ignore linting errors from Tensorflow's thoroughly broken import system
      from tensorflow.keras import backend as K, losses as k_losses  # pylint:disable=import-error
  
  _PARAMS = [(losses.GeneralizedLoss(), (2, 16, 16)),
             (losses.GradientLoss(), (2, 16, 16)),
             # TODO Make sure these output dimensions are correct
             (losses.GMSDLoss(**_KWARGS), (2, 1, 1)),
             # TODO Make sure these output dimensions are correct
 
</s> remove            (losses.GMSDLoss(), (2, 1, 1)),
 </s> add            (losses.GMSDLoss(**_KWARGS), (2, 1, 1)), </s> remove     from . import losses_tf as losses  #type:ignore # noqa
 </s> add  </s> remove     from . import losses_plaid as losses  # noqa
 </s> add  </s> remove     if get_backend() == "amd" and isinstance(loss_func, losses.GMSDLoss):
        pytest.skip("GMSD Loss is not currently compatible with PlaidML")
 </s> add 
_KWARGS = dict(input_dims=(64, 64)) if get_backend() == "amd" else {}
      # Ignore linting errors from Tensorflow's thoroughly broken import system
      from tensorflow.keras import backend as K, losses as k_losses  # pylint:disable=import-error
  
_KWARGS = dict(input_dims=(64, 64)) if get_backend() == "amd" else {}
  _PARAMS = [(losses.GeneralizedLoss(), (2, 16, 16)),
             (losses.GradientLoss(), (2, 16, 16)),
             # TODO Make sure these output dimensions are correct
             (losses.GMSDLoss(**_KWARGS), (2, 1, 1)),
             # TODO Make sure these output dimensions are correct
keep keep add keep keep keep keep keep
 Loss updates
  - Split Loss to own package
  - Add Focal Frequency Loss (Nvidia only)
  - Enable GMSD for AMD 
tests/lib/model/losses_test.py