diff options
-rw-r--r-- | ethosu/vela/operation.py | 6 |
1 files changed, 2 insertions, 4 deletions
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 |