|
|
|
@ -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):
|
|
|
|
|