tfa.register.register_keras_objects

View source on GitHub