fix (#102)

HSQ79815 · web-flow · commit 4a03f6ab5556 · 2022-10-28T17:48:05.000+08:00
Signed-off-by: haoshengqiang &lt;haoshengqiang79@163.com&gt;

Signed-off-by: haoshengqiang &lt;haoshengqiang79@163.com&gt;
diff --git a/onnxoptimizer/passes/eliminate_shape_gather.h b/onnxoptimizer/passes/eliminate_shape_gather.h
@@ -48,7 +48,7 @@ struct EliminateShapeGather final : public PredicateBasedPass {
 
     ONNX_ASSERT(indices_val < dims.size());
 
-    if (!dims[indices_val].is_int) {
+    if (!dims[indices_val].is_int || dims[indices_val].dim == -1) {
       return false;
     }
 

Original file line number	Diff line number	Diff line change
`@@ -48,7 +48,7 @@ struct EliminateShapeGather final : public PredicateBasedPass {`
`48`	`48`
`49`	`49`	`ONNX_ASSERT(indices_val < dims.size());`
`50`	`50`
`51`		`- if (!dims[indices_val].is_int) {`
	`51`	`+ if (!dims[indices_val].is_int \|\| dims[indices_val].dim == -1) {`
`52`	`52`	`return false;`
`53`	`53`	`}`
`54`	`54`