提交 55b85779 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Skip tests in float16 for sparse.

上级 79c49197
...@@ -2370,6 +2370,9 @@ class CastTester(utt.InferShapeTester): ...@@ -2370,6 +2370,9 @@ class CastTester(utt.InferShapeTester):
for format in sparse.sparse_formats: for format in sparse.sparse_formats:
for i_dtype in sparse.float_dtypes: for i_dtype in sparse.float_dtypes:
for o_dtype in tensor.float_dtypes: for o_dtype in tensor.float_dtypes:
if o_dtype == 'float16':
# Don't test float16 output.
continue
_, data = sparse_random_inputs( _, data = sparse_random_inputs(
format, format,
shape=(4, 7), shape=(4, 7),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论