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

Merge pull request !1279 from gongchen/fix_custom_op_case
pull/1279/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit aa1cee113d

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

Loading…
Cancel
Save