diff options
| author | Dwight Lidman <dwight.lidman@arm.com> | 2020-09-14 15:22:33 +0200 |
|---|---|---|
| committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-09-17 08:15:01 +0000 |
| commit | c3862c24baa9f141aa4f98b39dd6bd33353521fa (patch) | |
| tree | 6b7581ce36c400452b8f5d5c602cdfeabb116ca9 | |
| parent | 2f6f3790fba2e81594acd7ed927515e0367c150e (diff) | |
| download | ethos-u-vela-c3862c24baa9f141aa4f98b39dd6bd33353521fa.tar.gz | |
MLBEDSW-3008: Replace Split ops with Identity
A Split operation with num_splits=1 is essentially
a NOP. This commit adds a graph optimisation function
which replaces said Split ops with an Identity op for
later pruning.
Signed-off-by: Dwight Lidman <dwight.lidman@arm.com>
Change-Id: I0b0c535f214f54ee4c255662a18c37543bdc6d64
| -rw-r--r-- | ethosu/vela/graph_optimiser.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py index 1a6aaf10..a57ac82e 100644 --- a/ethosu/vela/graph_optimiser.py +++ b/ethosu/vela/graph_optimiser.py @@ -307,6 +307,15 @@ def fixup_resizebilinear(op, arch): return op +def convert_nop_split_to_identity(op, arch): + if op.type == "Split" and op.attrs.get("num_splits") == 1: + # the list comprehension should return a list with a single tensor + # if it shouldn't, remove_passthrough_tensor will fail appropriately + op.inputs = [i for i in op.inputs if i.shape == op.outputs[0].shape] + op.type = "Identity" + return op + + def fixup_fully_connected_input(op, arch): if op.type == "FullyConnectedAct": inp = op.inputs[0] @@ -956,6 +965,7 @@ def optimise_graph_a(nng, arch, verbose_graph=False): reorder_depthwise_weights, fixup_resizebilinear, fixup_bias_tensors, + convert_nop_split_to_identity, convert_mul_max_to_abs_or_lrelu, remove_unwanted_reshapes, convert_lrelu, |