From 3a26920b7cd302364d68830eb6e374311ce17f22 Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Thu, 21 Jan 2021 08:28:55 +0100 Subject: MLBEDSW-3772 Reshape removal -Removed reshapes in the original graph -Removed the addition of reshapes to the optimized graph -Reshapes with different ifm/ofm quantisation will remain Signed-off-by: Patrik Gustavsson Change-Id: I94862be53dac0d7434815e2aee5ca678228495f8 --- ethosu/vela/operation.py | 2 -- 1 file changed, 2 deletions(-) (limited to 'ethosu/vela/operation.py') diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 844f2985..342efd9d 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -629,7 +629,6 @@ class Operation: elif self.type == Op.StridedSlice: input_tens, begin_tens, end_tens, strides_tens = self.inputs outputs = self.outputs - out_tens = outputs[0] # Extract masks begin_mask = self.attrs["begin_mask"] @@ -641,7 +640,6 @@ class Operation: # shrink_axis_mask/new_axis_mask/ellipsis_mask is not supported by the Operation class but the operation # may have the attribute modified and handled in the graph optimization phase. assert shrink_axis_mask == new_axis_mask == ellipsis_mask == 0 - assert len(input_tens.shape) == len(out_tens.shape) offset_start = get_slice_offsets(input_tens.shape, begin_tens, begin_mask, is_begin=True) offset_end = get_slice_offsets(input_tens.shape, end_tens, end_mask, is_begin=False) elif self.type == Op.UnpackReshaped: -- cgit v1.2.1