Fix slice op size issue

If size[i] == -1, that means extract all elements of demension i subtract the begin[i] .
VeriSilicon · Oct 22, 2024 · 565a234 · 565a234
1 parent 35049c9
commit 565a234
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/op_map.cc b/op_map.cc
@@ -3052,8 +3052,8 @@ struct Slice : public OpMapperBase<EmptyStructPlaceholder> {
 
     for (int i = 0; i < size.size(); i++) {
       if (size[i] == -1) {  // If size[i] == -1, that means extract all
-                            // elements of demension i.
-        size[i] = input_tensor->GetShape()[i];
+                            // elements of demension i subtract the begin[i] .
+        size[i] = input_tensor->GetShape()[i] - begin[i];
       }
     }