yongwww commented on a change in pull request #4312:
URL: https://github.com/apache/incubator-tvm/pull/4312#discussion_r430613417
##########
File path: python/tvm/relay/op/transform.py
##########
@@ -607,31 +607,40 @@ def split(data, indices_or_sections, axis=0):
return TupleWrapper(_make.split(data, indices_or_sections, axis), ret_size)
-def strided_slice(data, begin, end, strides=None):
+def strided_slice(data, begin, end, strides=None, ignore_end=False):
"""Strided slice of an array.
Parameters
----------
data : relay.Expr
The source array to be sliced.
- begin: list of int
+ begin: relay.Expr or List[int]
The indices to begin with in the slicing.
- end: list of int
+ end: relay.Expr or List[int]
Indices indicating end of the slice.
- strides: list of int, optional
+ strides: relay.Expr or List[int], optional
Specifies the stride values, it can be negative in that case,
the input tensor will be reversed in that particular axis.
+ ignore_end: boolean, optional
+ Whether to ignore input end.
Review comment:
Added
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
