add depthwise conv attr

5 years ago · 3fbd0101cd
parent 0f42c66263
commit 3fbd0101cd
1 changed files with 1 additions and 0 deletions
--- a/mindspore/ops/operations/nn_ops.py
+++ b/mindspore/ops/operations/nn_ops.py
@ -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)