experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Do not test float16 for div

Browse Source 
Test float and int for resource scatter add

Fix indentation

Test half for scatter add

Remove tests
This commit is contained in:
WindQAQ 2020-07-10 03:02:45 -07:00
parent 5e28f0c5c8
commit 154ad2abb7
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/kernel_tests/scatter_ops_test.py
View File

@ -189,9 +189,11 @@ class ScatterTest(test.TestCase):
tf_scatter, tf_scatter,
repeat_indices=False, repeat_indices=False,
updates_are_scalar=False): updates_are_scalar=False):
vtypes = [np.float16, np.float32, np.float64] vtypes = [np.float32, np.float64]
if tf_scatter != state_ops.scatter_div: if tf_scatter != state_ops.scatter_div:
vtypes.append(np.int32) vtypes.append(np.int32)
# float16 is numerically unstable for div
vtypes.append(np.float16)
for vtype in vtypes: for vtype in vtypes:
for itype in (np.int32, np.int64): for itype in (np.int32, np.int64):