未验证 提交 b34b54db 编写于 作者: Z zhangxiaoci 提交者: GitHub

fix squeeze2/unsqueeze2 unittest, *test=kunlun (#43859)

上级 b3cf28f8
......@@ -38,6 +38,7 @@ class XPUTestSqueeze2Op(XPUOpTestWrapper):
def setUp(self):
self.op_type = "squeeze2"
self.__class__.op_type = "squeeze2"
self.use_mkldnn = False
self.init_dtype()
self.init_test_case()
......
......@@ -39,6 +39,7 @@ class XPUTestUnsqueeze2Op(XPUOpTestWrapper):
def setUp(self):
self.op_type = "unsqueeze2"
self.__class__.op_type = "unsqueeze2"
self.use_mkldnn = False
self.init_dtype()
self.init_test_case()
......@@ -117,6 +118,7 @@ class XPUTestUnsqueeze2Op(XPUOpTestWrapper):
def setUp(self):
self.op_type = "unsqueeze2"
self.__class__.op_type = "unsqueeze2"
self.use_mkldnn = False
self.init_dtype()
self.init_test_case()
......@@ -191,6 +193,7 @@ class XPUTestUnsqueeze2Op(XPUOpTestWrapper):
def setUp(self):
self.op_type = "unsqueeze2"
self.__class__.op_type = "unsqueeze2"
self.use_mkldnn = False
self.init_test_case()
self.init_dtype()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册