Browse Source

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

Merge pull request !1279 from gongchen/fix_custom_op_case
tags/v0.3.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
aa1cee113d
4 changed files with 1 additions and 1 deletions
  1. +0
    -0
      tests/st/ops/custom_ops_tbe/__init__.py
  2. +0
    -0
      tests/st/ops/custom_ops_tbe/conv2d_impl.py
  3. +1
    -1
      tests/st/ops/custom_ops_tbe/cus_conv2d.py
  4. +0
    -0
      tests/st/ops/custom_ops_tbe/test_conv.py

+ 0
- 0
tests/st/ops/custom_ops_tbe/__init__.py View File


tests/st/ops/custom_ops_tbe/cus_conv2d_impl.py → tests/st/ops/custom_ops_tbe/conv2d_impl.py View File


+ 1
- 1
tests/st/ops/custom_ops_tbe/cus_conv2d.py 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)


tests/st/ops/custom_ops_tbe/test_cus_conv.py → tests/st/ops/custom_ops_tbe/test_conv.py View File


Loading…
Cancel
Save