diff options
author | 2017-08-24 14:40:07 -0700 | |
---|---|---|
committer | 2017-08-24 14:43:50 -0700 | |
commit | 989713f265f5d395beb141fcf04b0ccb7aa68aab (patch) | |
tree | 0580c3c911c6da46759319554167a4a8bcdd7aa6 /tensorflow/python/kernel_tests/reduction_ops_test.py | |
parent | 105dd2aa22c7aaac401d2d8ed3131b5d81fdc227 (diff) |
Automated g4 rollback of changelist 166294015
PiperOrigin-RevId: 166396680
Diffstat (limited to 'tensorflow/python/kernel_tests/reduction_ops_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/reduction_ops_test.py | 18 |
1 files changed, 0 insertions, 18 deletions
diff --git a/tensorflow/python/kernel_tests/reduction_ops_test.py b/tensorflow/python/kernel_tests/reduction_ops_test.py index b98a04d72c..04ce99a4a6 100644 --- a/tensorflow/python/kernel_tests/reduction_ops_test.py +++ b/tensorflow/python/kernel_tests/reduction_ops_test.py @@ -175,24 +175,6 @@ class SumReductionTest(BaseReductionTest): np_arr = self._makeIncremental((2,) * rank, dtypes.int32) self._compareAllAxes(np_arr) - def testFloat16(self): - for rank in range(1, _MAX_RANK + 1): - np_arr = self._makeIncremental((2,) * rank, dtypes.float16) - self._compareAllAxes(np_arr) - - # test that mean doesn't overflow - # only on GPU, since it has the more accurate implementation - if not test.is_gpu_available(): - return - - arr = np.ones([68000], dtype=np.float16) - - with self.test_session(graph=ops.Graph(), use_gpu=True) as sess: - tf_arr = array_ops.constant(arr) - tf_mean = math_ops.reduce_mean(tf_arr, 0, False) - tf_out_mean = sess.run(tf_mean) - self.assertAllClose(tf_out_mean, 1.) - def testFloat32(self): for rank in range(1, _MAX_RANK + 1): np_arr = self._makeIncremental((2,) * rank, dtypes.float32) |