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

Add more test cases to CompressTester

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