Skip to content

Commit e720cd9

Browse files
committed
fix squeeze threshold
Signed-off-by: daquexian <daquexian566@gmail.com>
1 parent 6acd39c commit e720cd9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

onnxoptimizer/passes/fuse_add_bias_into_conv.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,8 @@ struct FuseAddBiasIntoConv final : public PredicateBasedPass {
4343
Node *squeeze = graph.create(k, 1);
4444
int opset_version = getOpsetVersion(graph);
4545
squeeze->addInput(input);
46-
int version_threshold = 11;
47-
if (opset_version <= version_threshold && opset_version != 0) {
46+
int version_threshold = 13;
47+
if (opset_version < version_threshold && opset_version != 0) {
4848
squeeze->is_(kaxes, std::move(axes));
4949
} else {
5050
Tensor t;

0 commit comments

Comments
 (0)