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

PiperOrigin-RevId: 288392606
google-deepmind · Jan 15, 2020 · b8cf445 · b8cf445
1 parent 8b47d2a
commit b8cf445
Show file tree

Hide file tree

Showing 22 changed files with 22 additions and 22 deletions.
diff --git a/cs_gan/cs.py b/cs_gan/cs.py
@@ -19,7 +19,7 @@
 import math
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from cs_gan import utils
 

diff --git a/cs_gan/gan.py b/cs_gan/gan.py
@@ -16,7 +16,7 @@
 import math
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 from cs_gan import utils
 

diff --git a/cs_gan/main.py b/cs_gan/main.py
@@ -21,7 +21,7 @@
 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

diff --git a/cs_gan/main_cs.py b/cs_gan/main_cs.py
@@ -22,7 +22,7 @@
 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

diff --git a/cs_gan/nets.py b/cs_gan/nets.py
@@ -17,7 +17,7 @@
 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
 
 

diff --git a/cs_gan/tests/gan_test.py b/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
 

diff --git a/cs_gan/utils.py b/cs_gan/utils.py
@@ -19,7 +19,7 @@
 
 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

diff --git a/curl/layers.py b/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
 

diff --git a/curl/model.py b/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

diff --git a/curl/training.py b/curl/training.py
@@ -22,7 +22,7 @@
 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
 

diff --git a/curl/utils.py b/curl/utils.py
@@ -20,7 +20,7 @@
 from __future__ import print_function
 
 from absl import logging
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import tensorflow_probability as tfp
 
 

diff --git a/hierarchical_probabilistic_unet/geco_utils.py b/hierarchical_probabilistic_unet/geco_utils.py
@@ -23,7 +23,7 @@
 
 import numpy as np
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 class MovingAverage(snt.AbstractModule):

diff --git a/hierarchical_probabilistic_unet/model_test.py b/hierarchical_probabilistic_unet/model_test.py
@@ -19,7 +19,7 @@
 from __future__ import print_function
 
 from model import HierarchicalProbUNet
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 _NUM_CLASSES = 2

diff --git a/hierarchical_probabilistic_unet/unet_utils.py b/hierarchical_probabilistic_unet/unet_utils.py
@@ -19,7 +19,7 @@
 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,

diff --git a/transporter/transporter.py b/transporter/transporter.py
@@ -21,7 +21,7 @@
 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
 

diff --git a/transporter/transporter_test.py b/transporter/transporter_test.py
@@ -19,7 +19,7 @@
 from __future__ import print_function
 
 from absl.testing import parameterized
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from transporter import transporter
 
 

diff --git a/tvt/losses.py b/tvt/losses.py
@@ -21,7 +21,7 @@
 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):

diff --git a/tvt/main.py b/tvt/main.py
@@ -28,7 +28,7 @@
 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

diff --git a/tvt/memory.py b/tvt/memory.py
@@ -23,7 +23,7 @@
 import collections
 
 import sonnet as snt
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 ReadInformation = collections.namedtuple(
     'ReadInformation', ('weights', 'indices', 'keys', 'strengths'))

diff --git a/tvt/rma.py b/tvt/rma.py
@@ -26,7 +26,7 @@
 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

diff --git a/unrestricted_advx/main.py b/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
 

diff --git a/unsupervised_adversarial_training/quick_eval_cifar.py b/unsupervised_adversarial_training/quick_eval_cifar.py
@@ -29,7 +29,7 @@
 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