Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-05-28 19:31:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source 
PiperOrigin-RevId: 288392606
This commit is contained in:
Deepmind Team
2020-01-06 23:51:12 +00:00
committed by Diego de Las Casas
parent 8b47d2a576
commit b8cf44502b
22 changed files with 22 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tvt/losses.py
+1 -1
View File
@@ -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
+1 -1
View File
@@ -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
+1 -1
View File
@@ -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
+1 -1
View File
@@ -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