Internal change

PiperOrigin-RevId: 299792626

official/r1/transformer/attention_layer.py

@@ -18,7 +18,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 class Attention(tf.layers.Layer):

official/r1/transformer/dataset.py

@@ -54,7 +54,7 @@ from __future__ import print_function
 import math
 import os
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from official.utils.misc import model_helpers
 

official/r1/transformer/embedding_layer.py

@@ -18,7 +18,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-import tensorflow as tf  # pylint: disable=g-bad-import-order
+import tensorflow.compat.v1 as tf  # pylint: disable=g-bad-import-order
 
 from official.r1.utils import tpu as tpu_utils
 

official/r1/transformer/ffn_layer.py

@@ -18,7 +18,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 class FeedFowardNetwork(tf.layers.Layer):

official/r1/transformer/schedule.py

@@ -20,7 +20,7 @@ from __future__ import print_function
 
 import math
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 _TRAIN, _EVAL = tf.estimator.ModeKeys.TRAIN, tf.estimator.ModeKeys.EVAL

official/r1/transformer/schedule_test.py

@@ -14,7 +14,7 @@
 # ==============================================================================
 """Test Transformer's schedule manager."""
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from official.r1.transformer import schedule
 

official/r1/transformer/transformer.py

@@ -22,7 +22,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from official.nlp.transformer import beam_search_v1 as beam_search
 from official.nlp.transformer import model_utils

official/r1/transformer/transformer_main.py

@@ -29,7 +29,7 @@ import tempfile
 from six.moves import xrange  # pylint: disable=redefined-builtin
 from absl import app as absl_app
 from absl import flags
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 # pylint: enable=g-bad-import-order
 
 from official.nlp.transformer import model_params

official/r1/transformer/translate.py

@@ -23,7 +23,7 @@ import os
 # pylint: disable=g-bad-import-order
 from absl import app as absl_app
 from absl import flags
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 # pylint: enable=g-bad-import-order
 
 from official.nlp.transformer.utils import tokenizer