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/test/testutil.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'ethosu/vela/test/testutil.py') diff --git a/ethosu/vela/test/testutil.py b/ethosu/vela/test/testutil.py index 96aeb7eb..02e01a51 100644 --- a/ethosu/vela/test/testutil.py +++ b/ethosu/vela/test/testutil.py @@ -113,14 +113,15 @@ def create_op_with_quant_tensors( return op -def create_op(op_type, inputs, output, attrs=None): +def create_op(op_type, inputs, output, attrs=None, set_ifm_ofm_shapes=True): op = Operation(op_type, output.name + "_op") for input in inputs: op.add_input_tensor(input) op.set_output_tensor(output) if attrs is not None: op.attrs = attrs - op.set_ifm_ofm_shapes() + if set_ifm_ofm_shapes: + op.set_ifm_ofm_shapes() return op -- cgit v1.2.1