未验证 提交 a39bd3ac 编写于 作者: L lijianshe02 提交者: GitHub

reduce the execution time of affine_channel op unittests (#26393)

* reduce the execution time of affine_channel op unittests
上级 eeeef957
......@@ -63,7 +63,7 @@ class TestAffineChannelOp(OpTest):
self.check_grad(['X'], 'Out', no_grad_set=set(['Scale', 'Bias']))
def init_test_case(self):
self.shape = [2, 100, 12, 12]
self.shape = [2, 100, 3, 3]
self.C = 100
self.layout = 'NCHW'
......@@ -102,7 +102,7 @@ class TestAffineChannelOpError(unittest.TestCase):
class TestAffineChannelNHWC(TestAffineChannelOp):
def init_test_case(self):
self.shape = [2, 12, 12, 100]
self.shape = [2, 3, 3, 100]
self.C = 100
self.layout = 'NHWC'
......@@ -115,7 +115,7 @@ class TestAffineChannelNHWC(TestAffineChannelOp):
class TestAffineChannel2D(TestAffineChannelOp):
def init_test_case(self):
self.shape = [8, 100]
self.shape = [2, 100]
self.C = 100
self.layout = 'NCHW'
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册