Hotfix to MKL-DNN pool2d tests (#22086)

release/1.7
Adam 5 years ago committed by Tao Luo
parent bf978fffe5
commit e1d666fb8b

@ -26,6 +26,9 @@ def create_test_mkldnn_use_ceil_class(parent):
def init_ceil_mode(self):
self.ceil_mode = True
def init_data_type(self):
self.dtype = np.float32
cls_name = "{0}_{1}".format(parent.__name__, "MKLDNNCeilModeCast")
TestMKLDNNPool2DUseCeilCase.__name__ = cls_name
globals()[cls_name] = TestMKLDNNPool2DUseCeilCase
@ -41,6 +44,9 @@ def create_test_mkldnn_class(parent):
def init_kernel_type(self):
self.use_mkldnn = True
def init_data_type(self):
self.dtype = np.float32
cls_name = "{0}_{1}".format(parent.__name__, "MKLDNNOp")
TestMKLDNNCase.__name__ = cls_name
globals()[cls_name] = TestMKLDNNCase
@ -78,6 +84,9 @@ class TestAsymPad(TestPool2D_Op):
def init_global_pool(self):
self.global_pool = False
def init_data_type(self):
self.dtype = np.float32
class TestAsymPadCase1(TestAsymPad):
def init_paddings(self):

Loading…
Cancel
Save