kevinthesun commented on a change in pull request #4312:
URL: https://github.com/apache/incubator-tvm/pull/4312#discussion_r432045059
##########
File path: python/tvm/relay/op/transform.py
##########
@@ -611,31 +611,41 @@ 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, slice_mode=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.
+ slice_mode: boolean, optional
+ Whether to ignore the negative elements in input end,
+ will slice to the end of data for the ignored element.
Review comment:
Provide more details about this attribute, such as end is actually slice
size and strides is ignored.
----------------------------------------------------------------
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:
us...@infra.apache.org
