recover reshape ut

This commit is contained in:
yao_yf 2020-07-22 14:30:12 +08:00
parent 3dbe872596
commit 1d3a06a3b0
1 changed files with 4 additions and 4 deletions

View File

@ -344,11 +344,11 @@ def test_reshape_net2_2():
reshape_net2(ReshapeNet2(((1, 8), (8, 2))))
def _test_reshape_net3_1():
def test_reshape_net3_1():
reshape_net2(ReshapeNet3(((1, 8), (8, 1))))
def _test_reshape_net3_2():
def test_reshape_net3_2():
reshape_net2(ReshapeNet3(((1, 8), (8, 2))))
@ -464,11 +464,11 @@ def test_reshape_common2_3():
reshape_common2(ParallelMode.SEMI_AUTO_PARALLEL, ReshapeNet2(((1, 8), (8, 2))))
def _test_reshape_common2_4():
def test_reshape_common2_4():
reshape_common2(ParallelMode.SEMI_AUTO_PARALLEL, ReshapeNet3(((1, 8), (8, 1))))
def _test_reshape_common2_5():
def test_reshape_common2_5():
reshape_common2(ParallelMode.SEMI_AUTO_PARALLEL, ReshapeNet3(((1, 8), (8, 2))))