aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/compatibility
diff options
context:
space:
mode:
authorGravatar imsheridan <xiaoyudong0512@gmail.com>2018-04-17 01:00:44 +0800
committerGravatar imsheridan <xiaoyudong0512@gmail.com>2018-04-17 01:00:44 +0800
commitc22d996c3d6a16db292bd3464b2ef7b91adae676 (patch)
tree131b65fe464f9f5bf2724117ca6f90f38680f859 /tensorflow/tools/compatibility
parent63c6562df68ade3a03481874a71b536a4e02b6f5 (diff)
Fix expand_dims of dims argument has been deprecated with axis
Diffstat (limited to 'tensorflow/tools/compatibility')
-rw-r--r--tensorflow/tools/compatibility/testdata/test_file_v0_11.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/tools/compatibility/testdata/test_file_v0_11.py b/tensorflow/tools/compatibility/testdata/test_file_v0_11.py
index 01f37d8768..40526d930c 100644
--- a/tensorflow/tools/compatibility/testdata/test_file_v0_11.py
+++ b/tensorflow/tools/compatibility/testdata/test_file_v0_11.py
@@ -94,7 +94,7 @@ class TestUpgrade(test_util.TensorFlowTestCase):
self.assertAllClose(
tf.reduce_logsumexp(a, [0, 1]).eval(), 6.45619344711)
self.assertAllEqual(
- tf.expand_dims([[1, 2], [3, 4]], dim=1).eval(),
+ tf.expand_dims([[1, 2], [3, 4]], axis=1).eval(),
[[[1, 2]], [[3, 4]]])
def testArgMinMax(self):