Explicitly replace "import tensorflow" with "tensorflow.compat.v1"

PiperOrigin-RevId: 288392606

cs_gan/cs.py

@@ -19,7 +19,7 @@ from __future__ import print_function
 import math
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from cs_gan import utils
 

cs_gan/gan.py

@@ -16,7 +16,7 @@ import collections
 import math
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from cs_gan import utils
 

cs_gan/main.py

@@ -21,7 +21,7 @@ import os
 from absl import app
 from absl import flags
 from absl import logging
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from cs_gan import file_utils
 from cs_gan import gan

cs_gan/main_cs.py

@@ -22,7 +22,7 @@ from absl import app
 from absl import flags
 from absl import logging
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_probability as tfp
 
 from cs_gan import cs

cs_gan/nets.py

@@ -17,7 +17,7 @@ import functools
 import re
 import numpy as np
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_gan as tfgan
 
 

cs_gan/tests/gan_test.py

@@ -14,7 +14,7 @@
 
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from cs_gan import gan
 

cs_gan/utils.py

@@ -19,7 +19,7 @@ import os
 
 from absl import logging
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_probability as tfp
 
 from cs_gan import nets

curl/layers.py

@@ -17,7 +17,7 @@
 
 from absl import logging
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 tfc = tf.compat.v1
 

curl/model.py

@@ -18,7 +18,7 @@
 from absl import logging
 import numpy as np
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_probability as tfp
 
 from curl import layers

curl/training.py

@@ -22,7 +22,7 @@ from absl import logging
 import numpy as np
 from sklearn import neighbors
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_datasets as tfds
 import tensorflow_probability as tfp
 

curl/utils.py

@@ -20,7 +20,7 @@ from __future__ import division
 from __future__ import print_function
 
 from absl import logging
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_probability as tfp
 
 

hierarchical_probabilistic_unet/geco_utils.py

@@ -23,7 +23,7 @@ from __future__ import print_function
 
 import numpy as np
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 class MovingAverage(snt.AbstractModule):

hierarchical_probabilistic_unet/model_test.py

@@ -19,7 +19,7 @@ from __future__ import division
 from __future__ import print_function
 
 from model import HierarchicalProbUNet
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 _NUM_CLASSES = 2

hierarchical_probabilistic_unet/unet_utils.py

@@ -19,7 +19,7 @@ from __future__ import division
 from __future__ import print_function
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 def res_block(input_features, n_channels, n_down_channels=None,

transporter/transporter.py

@@ -21,7 +21,7 @@ from __future__ import print_function
 import functools
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from tensorflow.contrib import framework as contrib_framework
 from tensorflow.contrib import layers as contrib_layers
 

transporter/transporter_test.py

@@ -19,7 +19,7 @@ from __future__ import division
 from __future__ import print_function
 
 from absl.testing import parameterized
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from transporter import transporter
 
 

tvt/losses.py

@@ -21,7 +21,7 @@ from __future__ import division
 from __future__ import print_function
 
 import six
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 def sum_time_average_batch(tensor, name=None):

tvt/main.py

@@ -28,7 +28,7 @@ from absl import logging
 import numpy as np
 from six.moves import range
 from six.moves import zip
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from tvt import batch_env
 from tvt import nest_utils

tvt/memory.py

@@ -23,7 +23,7 @@ from __future__ import print_function
 import collections
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 ReadInformation = collections.namedtuple(
     'ReadInformation', ('weights', 'indices', 'keys', 'strengths'))

tvt/rma.py

@@ -26,7 +26,7 @@ import numpy as np
 from six.moves import range
 from six.moves import zip
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import trfl
 
 from tvt import losses

unrestricted_advx/main.py

@@ -16,7 +16,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
 import tensorflow_hub as hub
 from unrestricted_advex import eval_kit
 

unsupervised_adversarial_training/quick_eval_cifar.py

@@ -29,7 +29,7 @@ from absl import flags
 import cleverhans
 from cleverhans import attacks
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from tensorflow.python.ops import math_grad
 import tensorflow_hub as hub