forked from mindspore-Ecosystem/mindspore
!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:
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…
Reference in New Issue