TOSA: Added support for ADD, SUB and MUL
Added support for ADD, SUB and MUL
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I52acdc126b16e2cf4096bcf7a77023ea7d204998
diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py
index 6c9fbce..9448749 100644
--- a/ethosu/vela/vela.py
+++ b/ethosu/vela/vela.py
@@ -85,7 +85,7 @@
)
output_tfl_filename = output_basename + "_vela.tflite"
- if input_name.endswith(".tflite") or input_name.endswith(".tosa"):
+ if input_name.endswith(".tflite"):
tflite_writer.write_tflite(nng, output_tfl_filename)
if input_name.endswith(".tosa"):
rawdata_writer.write_rawdata_output(nng, arch, output_basename)