  
  # Update layers into Keras custom objects
  for name, obj in inspect.getmembers(sys.modules[__name__]):
      if inspect.isclass(obj) and obj.__module__ == __name__:
          get_custom_objects().update({name: obj})
 
</s> remove         return dict(eps_std=self.eps_std,
                    seed=self.seed,
                    initialized=self.initialized)
 </s> add         return {"eps_std": self.eps_std,
                "seed": self.seed,
                "initialized": self.initialized} </s> remove if get_backend() == "amd":
    import keras
else:
    from tensorflow import keras

 </s> add  </s> remove from lib.utils import get_backend
 </s> add  </s> remove     assert ((_BACKEND in ("cpu", "directml") and lib == "tensorflow")
            or (_BACKEND == "amd" and lib == "plaidml"))
 </s> add     assert _BACKEND in ("cpu", "directml") and lib == "tensorflow"
        keras.utils.get_custom_objects().update({name: obj})
  
  # Update layers into Keras custom objects
  for name, obj in inspect.getmembers(sys.modules[__name__]):
      if inspect.isclass(obj) and obj.__module__ == __name__:
        keras.utils.get_custom_objects().update({name: obj})
keep keep keep keep replace
 Remove plaidML support (#1325)

* Remove PlaidML reference from readme files

* Remove AMD option from installers

* remove amd requirements and update setup.py

* remove plaidml test from CI workflow

* gpustats: remove plaidml backend

* plaid removals:
  - faceswap.py - python version check
  - setup.cfg - plaidml typing ignore
  - lib.keras_utils - All plaid code
  - lib.launcher.py - All plaidml checks and configuration

* remove tf2.2 specific code from GUI event reader

* lib.model - remove all plaidml implementations

* plugins.extract - remove plaidml code

* plugins.train remove plaidml code

* lib.convert - remove plaidml code

* tools.model: remove plaidml code

* Remove plaidML tests from unit tests

* remove plaidml_utils and docsting cleanups

* Remove plaidML refs from configs

* fix keras imports 
lib/model/layers.py