Fixed strided_slice size (apache#7659)

Co-authored-by: Akira Maruoka <akira.maruoka@fixstars.com>
neo-ai · May 11, 2021 · c698a52 · c698a52
1 parent ab3e88a
commit c698a52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py
@@ -3093,7 +3093,7 @@ def convert_detection_postprocess(self, op):
         valid_count = ret[0]
         # keep only the top 'max_detections' rows
         ret = _op.strided_slice(
-            ret[1], [0, 0, 0], [batch_size, custom_options["max_detections"], anchor_boxes]
+            ret[1], [0, 0, 0], [batch_size, custom_options["max_detections"], 6]
         )
         # the output needs some reshaping to match tflite
         ret = _op.split(ret, 6, axis=2)