diff options
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): |