pnnx skip dynamic size evaluation (#4291)

Tencent · Oct 20, 2022 · 8edc03c · 8edc03c
1 parent 8eab5ea
commit 8edc03c
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/tools/pnnx/src/pass_level5/eval_expression.cpp b/tools/pnnx/src/pass_level5/eval_expression.cpp
@@ -126,7 +126,17 @@ static std::string eval_expression(const Operator* op)
                 {
                     int bi = std::stoi(b);
                     int r = op->inputs[input_index]->shape[bi];
-                    exprstack.push(std::to_string(r));
+                    if (r == -1)
+                    {
+                        // do not evaluate dynamic size info as -1
+                        // just keep the size expression
+                        std::string r = std::string("size(") + a + "," + b + ")";
+                        exprstack.push(r);
+                    }
+                    else
+                    {
+                        exprstack.push(std::to_string(r));
+                    }
                 }
             }
             else