aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/vela.py
diff options
context:
space:
mode:
authorJonas Ohlsson <jonas.ohlsson@arm.com>2021-09-01 15:57:21 +0200
committerJonas Ohlsson <jonas.ohlsson@arm.com>2021-09-15 10:48:08 +0100
commit0957e3ef4b94f17efb67429c88bab8ba650f78e8 (patch)
treef3e2600367bc7c89145657023b45b9dde2c316c2 /ethosu/vela/vela.py
parent1a7527cd4ad56b49f120b10dc5e87a1e8f5a8122 (diff)
downloadethos-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.py10
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",