diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index d2f2806..f36e61c 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -250,7 +250,6 @@
             shrink_axis_mask = self.attrs["shrink_axis_mask"]
             # TODO: Either extend this to support these different masks or check
             # for this at an earlier stage and place the op on Cpu if needed
-            assert begin_mask == end_mask
             assert new_axis_mask == ellipsis_mask == 0
             # shrink_axis_mask is not supported by the Operation class but the operation
             # may have the attribute modified and handled in the graph optimization phase.
@@ -258,18 +257,17 @@
             assert len(input_tens.shape) == len(out_tens.shape)
 
             for idx in range(len(input_tens.shape)):
-                # If the i:th bit in begin_mask is set then the value on begin[i] should be ignored
-                if (begin_mask & (1 << idx)) == 0:
-                    # Check if the op should slice in dimension idx
-                    if end_tens.values[idx] != input_tens.shape[idx] or (
-                        end_tens.values[idx] == input_tens.shape[idx] and begin_tens.values[idx] != 0
-                    ):
+                # Check if slicing is needed in this axis
+                if end_tens.values[idx] != input_tens.shape[idx] or (
+                    end_tens.values[idx] == input_tens.shape[idx] and begin_tens.values[idx] != 0
+                ):
+                    # If the i:th bit in begin_mask is set then the value on begin[i] should be ignored
+                    if (begin_mask & (1 << idx)) == 0:
                         offset_start[idx] = begin_tens.values[idx]
-                        offset_end[idx] = end_tens.values[idx]
 
-                else:
-                    # Don't slice in this axis, instead use fullest possible range
-                    continue
+                    # If the i:th bit in end_mask is set then the value on end[i] should be ignored
+                    if (end_mask & (1 << idx)) == 0:
+                        offset_end[idx] = end_tens.values[idx]
 
         elif self.type == "UnpackReshaped":
             # Requires fixup_unpack_output to be called before this point