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

* reduce the execution time of affine_channel op unittests
revert-24895-update_cub
lijianshe02 5 years ago committed by GitHub
parent eeeef957c7
commit a39bd3acc9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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'

Loading…
Cancel
Save