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 162f1f7Copy full SHA for 162f1f7
python/tvm/relay/op/contrib/tensorrt.py
@@ -709,7 +709,8 @@ 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)):
+ length: int = len(attrs.axes) if attrs.axes is not None else len(args[0].checked_type.shape)
713
+ for i in range(0, length):
714
begin = int(attrs.begin[i])
715
if attrs.slice_mode == "end":
716
end = (
0 commit comments