提交 da637d50 authored 作者: Seon-Wook Park's avatar Seon-Wook Park

Add axis=-1 case to CompressTester

上级 4a0e04dc
...@@ -346,16 +346,19 @@ class SqueezeTester(utt.InferShapeTester): ...@@ -346,16 +346,19 @@ class SqueezeTester(utt.InferShapeTester):
class CompressTester(utt.InferShapeTester): class CompressTester(utt.InferShapeTester):
axis_list = [None, axis_list = [None,
-1,
0, 0,
0, 0,
0, 0,
1] 1]
cond_list = [[1, 0, 1, 0, 0, 1], cond_list = [[1, 0, 1, 0, 0, 1],
[0, 1, 1, 0],
[0, 1, 1, 0], [0, 1, 1, 0],
[], [],
[0, 0, 0, 0], [0, 0, 0, 0],
[1, 1, 0, 1, 0]] [1, 1, 0, 1, 0]]
shape_list = [(2, 3), shape_list = [(2, 3),
(4, 3),
(4, 3), (4, 3),
(4, 3), (4, 3),
(4, 3), (4, 3),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论