Internal change

PiperOrigin-RevId: 323627192

Internal change
PiperOrigin-RevId: 323627192
322d2eb7 · A. Unique TensorFlower · 006df625 · 322d2eb7
Commit 322d2eb7 authored Jul 28, 2020 by A. Unique TensorFlower
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

official/nlp/xlnet/xlnet_modeling.py official/nlp/xlnet/xlnet_modeling.py +2 -2

No files found.
--- a/official/nlp/xlnet/xlnet_modeling.py
+++ b/official/nlp/xlnet/xlnet_modeling.py
@@ -15,13 +15,13 @@
 """Keras layers of XLNet model in TF 2.0."""

 import copy
-import functools

 import tensorflow as tf
 from official.nlp.xlnet import data_utils


-gelu = functools.partial(tf.keras.activations.gelu, approximate=True)
+def gelu(x):
+  return tf.keras.activations.gelu(x, approximate=True)


 def rel_shift(x, klen=-1):