Internal change

PiperOrigin-RevId: 323627192

Internal change
PiperOrigin-RevId: 323627192
260fa059 · A. Unique TensorFlower · a0a0dbde · 260fa059
Commit 260fa059 authored Jul 28, 2020 by A. Unique TensorFlower
Hide 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):