From: Yong Tang Date: Mon, 12 Feb 2018 12:07:24 +0000 (+0000) Subject: Fix some keep_dims warnings in math_ops_tests X-Git-Tag: upstream/v1.7.0~204^2 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=a09c5af9c94e2799ffe6162a342808742e88ab11;p=platform%2Fupstream%2Ftensorflow.git Fix some keep_dims warnings in math_ops_tests Signed-off-by: Yong Tang --- diff --git a/tensorflow/python/ops/math_ops_test.py b/tensorflow/python/ops/math_ops_test.py index bd26ff6..d314124 100644 --- a/tensorflow/python/ops/math_ops_test.py +++ b/tensorflow/python/ops/math_ops_test.py @@ -105,7 +105,7 @@ class LogSumExpTest(test_util.TensorFlowTestCase): for dtype in [np.float16, np.float32, np.double]: x_np = np.random.rand(5, 5).astype(dtype) with self.test_session(use_gpu=True): - y_tf_np = math_ops.reduce_logsumexp(x_np, keep_dims=True).eval() + y_tf_np = math_ops.reduce_logsumexp(x_np, keepdims=True).eval() self.assertEqual(y_tf_np.ndim, x_np.ndim) y_np = log(np.sum(exp(x_np), keepdims=True)) self.assertAllClose(y_tf_np, y_np)