2 parents 2c0cb01 + 8b04689 commit abe5aa4Copy full SHA for abe5aa4
1 file changed
onnx2tf/ops/Conv.py
@@ -280,6 +280,10 @@ def make_node(
280
print(error_msg)
281
assert False, error_msg
282
283
+ # Refresh after the NCHW->NHWC workaround above: the depthwise test below reads
284
+ # input_tensor_shape[-1], so a stale shape turns depthwise convs into grouped convs.
285
+ input_tensor_shape = input_tensor.shape
286
+
287
# DepthwiseConv2D
288
# 1. rank=4
289
# 2. group>1
0 commit comments