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

…r TF2.x migration PiperOrigin-RevId: 295207815
Teora · Feb 14, 2020 · 6df7081 · 6df7081
1 parent 1d69caf
commit 6df7081
Show file tree

Hide file tree

Showing 12 changed files with 12 additions and 12 deletions.
diff --git a/axial/config_imagenet32.py b/axial/config_imagenet32.py
@@ -17,7 +17,7 @@
 from __future__ import division
 from __future__ import print_function
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 def get_config():

diff --git a/axial/config_imagenet64.py b/axial/config_imagenet64.py
@@ -17,7 +17,7 @@
 from __future__ import division
 from __future__ import print_function
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 def get_config():

diff --git a/axial/utils.py b/axial/utils.py
@@ -18,7 +18,7 @@
 from __future__ import print_function
 
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from tensorflow.python.tpu import tpu_function
 
 

diff --git a/depth_from_video_in_the_wild/consistency_losses.py b/depth_from_video_in_the_wild/consistency_losses.py
@@ -19,7 +19,7 @@
 from __future__ import division
 from __future__ import print_function
 
-import tensorflow as tf  # tf
+import tensorflow.compat.v1 as tf  # tf
 from depth_from_video_in_the_wild import transform_utils
 from tensorflow.contrib import resampler as contrib_resampler
 

diff --git a/depth_from_video_in_the_wild/depth_prediction_net.py b/depth_from_video_in_the_wild/depth_prediction_net.py
@@ -19,7 +19,7 @@
 """
 
 import numpy as np
-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/depth_from_video_in_the_wild/model.py b/depth_from_video_in_the_wild/model.py
@@ -24,7 +24,7 @@
 
 from absl import logging
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from depth_from_video_in_the_wild import consistency_losses
 from depth_from_video_in_the_wild import depth_prediction_net
 from depth_from_video_in_the_wild import motion_prediction_net

diff --git a/protoattend/main_protoattend.py b/protoattend/main_protoattend.py
@@ -27,7 +27,7 @@
 import model
 import numpy as np
 from options import FLAGS
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 import utils
 
 # GPU options

diff --git a/protoattend/model.py b/protoattend/model.py
@@ -22,7 +22,7 @@
 import input_data
 import numpy as np
 from options import FLAGS
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 # Building blocks of the network

diff --git a/protoattend/options.py b/protoattend/options.py
@@ -15,7 +15,7 @@
 
 """Input flags."""
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 flags = tf.app.flags
 FLAGS = flags.FLAGS

diff --git a/tabnet/data_helper_covertype.py b/tabnet/data_helper_covertype.py
@@ -15,7 +15,7 @@
 
 """Data helper function for the Forest Covertype dataset."""
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 # Dataset size
 # N_TRAIN_SAMPLES = 309871

diff --git a/tabnet/data_helper_poker.py b/tabnet/data_helper_poker.py
@@ -15,7 +15,7 @@
 
 """Data helper function for the Poker dataset."""
 
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 # Dataset size
 # N_TRAIN_SAMPLES = 25010

diff --git a/tabnet/tabnet_model.py b/tabnet/tabnet_model.py
@@ -20,7 +20,7 @@
 from __future__ import print_function
 
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 def glu(act, n_units):