Skip to content

GitLab

Commit 7f824479 authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Clean up attention layer imports 

PiperOrigin-RevId: 328085299
parent 00c7ea79
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 5 deletions
official/nlp/modeling/layers/attention.py
View file @ 7f824479
...@@ -15,17 +15,12 @@ ...@@ -15,17 +15,12 @@
# ============================================================================== # ==============================================================================
"""Keras-based attention layer.""" """Keras-based attention layer."""
# pylint: disable=g-classes-have-attributes # pylint: disable=g-classes-have-attributes
import math import math
import string
import tensorflow as tf import tensorflow as tf
EinsumDense = tf.keras.layers.experimental.EinsumDense EinsumDense = tf.keras.layers.experimental.EinsumDense
_CHR_IDX = string.ascii_lowercase
MultiHeadAttention = tf.keras.layers.MultiHeadAttention MultiHeadAttention = tf.keras.layers.MultiHeadAttention
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment