From 154ad2abb7748624633b959eaab0b559aaeaf58e Mon Sep 17 00:00:00 2001
From: WindQAQ <windqaq@gmail.com>
Date: Fri, 10 Jul 2020 03:02:45 -0700
Subject: [PATCH] Do not test float16 for div
Test float and int for resource scatter add
Fix indentation
Test half for scatter add
Remove tests
---
tensorflow/python/kernel_tests/scatter_ops_test.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/tensorflow/python/kernel_tests/scatter_ops_test.py b/tensorflow/python/kernel_tests/scatter_ops_test.py
index e8b23d1b686..b9206bf3221 100644
--- a/tensorflow/python/kernel_tests/scatter_ops_test.py
+++ b/tensorflow/python/kernel_tests/scatter_ops_test.py
@@ -189,9 +189,11 @@ class ScatterTest(test.TestCase):
tf_scatter,
repeat_indices=False,
updates_are_scalar=False):
- vtypes = [np.float16, np.float32, np.float64]
+ vtypes = [np.float32, np.float64]
if tf_scatter != state_ops.scatter_div:
vtypes.append(np.int32)
+ # float16 is numerically unstable for div
+ vtypes.append(np.float16)
for vtype in vtypes:
for itype in (np.int32, np.int64):