Internal change

PiperOrigin-RevId: 345773699

Internal change
PiperOrigin-RevId: 345773699
c70aea2b · A. Unique TensorFlower · d74baa35 · c70aea2b
Commit c70aea2b authored Dec 04, 2020 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 0 deletions

official/nlp/keras_nlp/layers/on_device_embedding.py official/nlp/keras_nlp/layers/on_device_embedding.py +8 -0

No files found.
--- a/official/nlp/keras_nlp/layers/on_device_embedding.py
+++ b/official/nlp/keras_nlp/layers/on_device_embedding.py
@@ -96,3 +96,11 @@ class OnDeviceEmbedding(tf.keras.layers.Layer):
    if self._scale_factor:
      embeddings *= self._scale_factor
    return embeddings
+
+  @property
+  def vocab_size(self):
+    return self._vocab_size
+
+  @property
+  def embedding_width(self):
+    return self._embedding_width