aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools
diff options
context:
space:
mode:
authorGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-24 09:31:02 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-24 09:31:07 -0700
commitaf959c2f25e29ff1a76d8ad2a8100780e1bca8cf (patch)
treec393bef2c2fb02831dd4c99a20510502c4920f0b /tensorflow/tools
parente2ce9787d9927e4a6574e6ac4606a47712320170 (diff)
parent413ac36f33deb0c354dd687963d2410eab048970 (diff)
Merge pull request #18567 from imsheridan:fix_expand_dims
PiperOrigin-RevId: 214278672
Diffstat (limited to 'tensorflow/tools')
-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 35a74c9664..68ba7a2630 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):