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

Merge pull request !1279 from gongchen/fix_custom_op_case
This commit is contained in:
mindspore-ci-bot 2020-05-20 15:21:31 +08:00 committed by Gitee
commit aa1cee113d
4 changed files with 1 additions and 1 deletions

View File

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