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