fix wrong unsqueeze on opset 12

daquexian · daquexian · commit 156fc4d22937 · 2023-06-26T16:41:04.000+08:00
Signed-off-by: daquexian &lt;daquexian566@gmail.com&gt;
diff --git a/onnxoptimizer/passes/fuse_bn_into_conv.h b/onnxoptimizer/passes/fuse_bn_into_conv.h
@@ -127,7 +127,7 @@ struct FuseBNIntoConv final : public PredicateBasedPass {
     for (int i = 1; i < conv_W.sizes().size(); ++i) {
       insert_dims.push_back(i);
     }
-    if (getOpsetVersion(graph) > 11) {
+    if (getOpsetVersion(graph) >= 13) {
       Tensor shape_s_t;
       shape_s_t.elem_type() = ONNX_NAMESPACE::TensorProto_DataType_INT64;
       shape_s_t.sizes().push_back(insert_dims.size());

Original file line number	Diff line number	Diff line change
`@@ -127,7 +127,7 @@ struct FuseBNIntoConv final : public PredicateBasedPass {`
`127`	`127`	`for (int i = 1; i < conv_W.sizes().size(); ++i) {`
`128`	`128`	`insert_dims.push_back(i);`
`129`	`129`	`}`
`130`		`- if (getOpsetVersion(graph) > 11) {`
	`130`	`+ if (getOpsetVersion(graph) >= 13) {`
`131`	`131`	`Tensor shape_s_t;`
`132`	`132`	`shape_s_t.elem_type() = ONNX_NAMESPACE::TensorProto_DataType_INT64;`
`133`	`133`	`shape_s_t.sizes().push_back(insert_dims.size());`