Clean up attention layer imports

PiperOrigin-RevId: 328085299

Clean up attention layer imports
PiperOrigin-RevId: 328085299
7f824479 · Hongkun Yu · A. Unique TensorFlower · 00c7ea79 · 7f824479
Commit 7f824479 authored Aug 23, 2020 by Hongkun Yu Committed by A. Unique TensorFlower Aug 23, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 5 deletions

official/nlp/modeling/layers/attention.py official/nlp/modeling/layers/attention.py +0 -5

No files found.
--- a/official/nlp/modeling/layers/attention.py
+++ b/official/nlp/modeling/layers/attention.py
@@ -15,17 +15,12 @@
 # ==============================================================================
 """Keras-based attention layer."""
 # pylint: disable=g-classes-have-attributes
-
 import math
-import string

 import tensorflow as tf


 EinsumDense = tf.keras.layers.experimental.EinsumDense
-_CHR_IDX = string.ascii_lowercase
-
-
 MultiHeadAttention = tf.keras.layers.MultiHeadAttention