MLBEDSW-7059: Updated offset calculation for Slice
Corrected offset calculation for operator Slice. All values
in tensor begin and tensor size must be used to calculate the
offset range in order to read the correct data.
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Change-Id: Ic463d8f72a2167f8129109b8dcf005f034cce6ed
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index 4a56f1f..b4d0e48 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -762,10 +762,8 @@
offset_end = [0] * len(input_tens.shape)
for idx in range(len(begin_tens.values)):
- # Check if the op should slice in dimension idx
- if size_tens.values[idx] != input_tens.shape[idx]:
- offset_start[idx] = begin_tens.values[idx]
- offset_end[idx] = size_tens.values[idx] + offset_start[idx]
+ offset_start[idx] = begin_tens.values[idx]
+ offset_end[idx] = size_tens.values[idx] + offset_start[idx]
elif self.type == Op.StridedSlice:
input_tens, begin_tens, end_tens, strides_tens = self.inputs