diff options
author | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2021-09-01 15:57:21 +0200 |
---|---|---|
committer | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2021-09-15 10:48:08 +0100 |
commit | 0957e3ef4b94f17efb67429c88bab8ba650f78e8 (patch) | |
tree | f3e2600367bc7c89145657023b45b9dde2c316c2 /ethosu/vela/vela.py | |
parent | 1a7527cd4ad56b49f120b10dc5e87a1e8f5a8122 (diff) | |
download | ethos-u-vela-0957e3ef4b94f17efb67429c88bab8ba650f78e8.tar.gz |
MLBEDSW-5102 Update removal of memory only operators
Memory only operators such as Reshape, Squeeze and ExpandDims are
removed in the graph optimiser step.
- Added semantic check that memory only operators have same
quantisation parameters on ifm/ofm.
- Added support for the ExpandDims operator.
- Addition and cleanup of related unit tests.
- Removed TOSA from the generated SUPPORTED_OPS.md documentation.
Signed-off-by: Jonas Ohlsson <jonas.ohlsson@arm.com>
Change-Id: If848d8afc58c18806e10997ed94e4dae83f30879
Diffstat (limited to 'ethosu/vela/vela.py')
-rw-r--r-- | ethosu/vela/vela.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py index 94487499..8a808276 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -167,6 +167,10 @@ def print_subgraph_io_summary(nng): def generate_supported_ops(): + # Exclude network type from generation by adding value to exclude list. + # To easily exclude NetworkType from generated documentation. + exclude_generation_network_type_value = [NetworkType.TOSA.value] + lines = [ "# Supported Ops", "", @@ -180,11 +184,17 @@ def generate_supported_ops(): ] for network_type in NetworkType: + if network_type.value in exclude_generation_network_type_value: + continue + lines += [ f"- [{network_type.name}](#{network_type.name.lower()}-summary-table)", ] for network_type in NetworkType: + if network_type.value in exclude_generation_network_type_value: + continue + lines += [ "", f"## {network_type.name} Summary Table", |