Public export.

PiperOrigin-RevId: 331634511

official/nlp/keras_nlp/layers/on_device_embedding.py

-# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2020 The TensorFlow Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

official/nlp/keras_nlp/layers/position_embedding.py

@@ -17,7 +17,7 @@
 import tensorflow as tf
 
 
-@tf.keras.utils.register_keras_serializable(package="Text")
+@tf.keras.utils.register_keras_serializable(package="keras_nlp")
 class PositionEmbedding(tf.keras.layers.Layer):
   """Creates a positional embedding.
 

official/nlp/keras_nlp/layers/transformer_encoder_block.py

@@ -14,11 +14,10 @@
 # ==============================================================================
 """Keras-based TransformerEncoder block layer."""
 
-# Import libraries
 import tensorflow as tf
 
 
-@tf.keras.utils.register_keras_serializable(package="Text")
+@tf.keras.utils.register_keras_serializable(package="keras_nlp")
 class TransformerEncoderBlock(tf.keras.layers.Layer):
   """TransformerEncoderBlock layer.