commit c74682cfd27eb2c203ce4486e712916c45da9881
author Patrik Gustavsson <patrik.gustavsson@arm.com> Tue Aug 17 14:26:38 2021 +0200
committer Patrik Gustavsson <patrik.gustavsson@arm.com> Fri Sep 03 13:33:01 2021 +0200
tree 82ff1cefd0ce06d6072f0b1231802e7afa803b1a
parent 5e5a7847b8fc1eb261c7561f44585d2f6b524df3

TOSA: Support for AVGPOOL, MAXPOOL and CONV2D

Added support for
-AVGPOOL and CONV2D with TFLite correspondence
-MAXPOOL
-additional support for replacing RESCALE ops with avgpool.

No support for breaking down tensors over the
size supported by NPU.

Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I1d2aa50ac30a26283b3e6f1fe88cba1544b7c189

ethosu/vela/vela.py

diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py
index 7400b8e..9448749 100644
--- a/ethosu/vela/vela.py
+++ b/ethosu/vela/vela.py
@@ -87,7 +87,7 @@
     output_tfl_filename = output_basename + "_vela.tflite"
     if input_name.endswith(".tflite"):
         tflite_writer.write_tflite(nng, output_tfl_filename)
-    elif input_name.endswith(".tosa"):
+    if input_name.endswith(".tosa"):
         rawdata_writer.write_rawdata_output(nng, arch, output_basename)
 
     if enable_debug_db: