Browse Source

add depthwise conv attr

tags/v0.3.0-alpha
wangcong 5 years ago
parent
commit
3fbd0101cd
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ops/operations/nn_ops.py

+ 1
- 0
mindspore/ops/operations/nn_ops.py View File

@@ -862,6 +862,7 @@ class DepthwiseConv2dNative(PrimitiveWithInfer):
self.channel_multiplier = validator.check_integer("channel_multiplier", channel_multiplier, 0, Rel.GT,
self.name)
self.group = validator.check_integer("group", group, 0, Rel.GT, self.name)
self.add_prim_attr('offset_a', 0)

def infer_shape(self, x_shape, w_shape):
validator.check_integer("weight rank", len(w_shape), 4, Rel.EQ, self.name)


Loading…
Cancel
Save