onnx2tf 1.29.1__py3-none-any.whl → 1.29.2__py3-none-any.whl

onnx2tf/__init__.py

@@ -1,3 +1,3 @@
 from onnx2tf.onnx2tf import convert, main
 
-__version__ = '1.29.1'
+__version__ = '1.29.2'

onnx2tf/ops/Conv.py

@@ -370,6 +370,20 @@ def make_node(
             )
 
     def depth_conv_bias(input_tensor, input_weights, pad_mode, strides, dilations, input_bias):
+        # tf.nn.depthwise_conv2d uses a different output shape when dilation>1 and stride>1.
+        # Emulate stride>1 by running stride=1 then subsampling to match ONNX.
+        if pad_mode == 'VALID' \
+            and max(dilations) > 1 \
+            and any(s > 1 for s in strides[1:-1]):
+            conv = tf.nn.depthwise_conv2d(
+                input=input_tensor,
+                filter=input_weights,
+                padding=pad_mode,
+                strides=[1, 1, 1, 1],
+                dilations=dilations,
+            )
+            conv = conv[:, ::strides[1], ::strides[2], :]
+            return tf.add(conv, input_bias)
         return \
             tf.add(
                 tf.nn.depthwise_conv2d(
@@ -438,6 +452,19 @@ def make_node(
             )
 
     def depth_conv_nobias(input_tensor, input_weights, pad_mode, strides, dilations):
+        # tf.nn.depthwise_conv2d uses a different output shape when dilation>1 and stride>1.
+        # Emulate stride>1 by running stride=1 then subsampling to match ONNX.
+        if pad_mode == 'VALID' \
+            and max(dilations) > 1 \
+            and any(s > 1 for s in strides[1:-1]):
+            conv = tf.nn.depthwise_conv2d(
+                input=input_tensor,
+                filter=input_weights,
+                padding=pad_mode,
+                strides=[1, 1, 1, 1],
+                dilations=dilations,
+            )
+            return conv[:, ::strides[1], ::strides[2], :]
         return \
             tf.nn.depthwise_conv2d(
                 input=input_tensor,