提交 aa1cee11 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!1279 test(custom_op): clean up custom op test case.

Merge pull request !1279 from gongchen/fix_custom_op_case
......@@ -103,7 +103,7 @@ class Cus_Conv2D(PrimitiveWithInfer):
raise ValueError('Attr \'kernel_size\' of \'Conv2D\' Op passed '
+ str(self.kernel_size) + ', should be a int or tuple and equal to or greater than 1.')
self.stride = validator.check_integer('stride', stride, 1, Rel.GE)
from .cus_conv2d_impl import Cus_Conv2D
from conv2d_impl import Cus_Conv2D
def infer_shape(self, x_shape, w_shape):
validator.check_integer("weight_shape", len(w_shape), 4, Rel.EQ)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册