diff options
author | 2018-09-24 09:31:02 -0700 | |
---|---|---|
committer | 2018-09-24 09:31:07 -0700 | |
commit | af959c2f25e29ff1a76d8ad2a8100780e1bca8cf (patch) | |
tree | c393bef2c2fb02831dd4c99a20510502c4920f0b /tensorflow/tools | |
parent | e2ce9787d9927e4a6574e6ac4606a47712320170 (diff) | |
parent | 413ac36f33deb0c354dd687963d2410eab048970 (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.py | 2 |
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): |