We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0e046da commit 2c9772aCopy full SHA for 2c9772a
python/tvm/relay/op/contrib/tensorrt.py
@@ -709,7 +709,9 @@ def strided_slice_checker(
709
if any([x is not None and x <= 0 for x in attrs.strides]):
710
logger.info(f"{op_name}: stride must be positive")
711
return False
712
- for i in range(0, len(args[0].checked_type.shape)):
+ if len(attrs.axes) > len(args[0].checked_type.shape):
713
+ logger.info(f"{op_name}: number of indexed axes is higher than input tensor rank")
714
+ for i in range(0, len(attrs.axes)):
715
begin = int(attrs.begin[i])
716
if attrs.slice_mode == "end":
717
end = (
0 commit comments