From 0b799e465b7b0ce8c5ea4f246a61773f607ebf74 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Johan=20Alfv=C3=A9n?= Date: Tue, 25 Oct 2022 16:22:58 +0200 Subject: 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 Change-Id: Ic463d8f72a2167f8129109b8dcf005f034cce6ed --- ethosu/vela/operation.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) (limited to 'ethosu') diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 4a56f1f0..b4d0e48a 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -762,10 +762,8 @@ class Operation: 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 -- cgit v1.2.1