commit 9fbc4913fe5dd0e6090532963f8612449936d994
author Jacob Bohlin <jacob.bohlin@arm.com> Mon Jun 29 11:58:50 2020 +0200
committer tim.hall <tim.hall@arm.com> Fri Jul 10 14:20:20 2020 +0000
tree 01fa3f202116cc4b767077f5e0a3977cf2185559
parent 0b9ca78dbde3f5c29f368577de1f621845e711a2

MLBEDSW-1497: Add Quantize operator support

Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: Iaf4d7ab9c32b0d783072c5f131a61bfebe77cc16

ethosu/vela/supported_operators.py

diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py
index 55e718e..3bf46a9 100644
--- a/ethosu/vela/supported_operators.py
+++ b/ethosu/vela/supported_operators.py
@@ -66,6 +66,8 @@
             | set(("ConcatSliceWrite"))
             # bias add and batch norm
             | set(("QuantizedBiasAdd", "Requantize", "QuantizedBatchNorm", "BiasAdd", "FusedBatchNorm"))
+            # Quantization
+            | set(("Quantize",))
         )
         self.split_ops = set(("Split", "SplitV", "StridedSlice", "Slice", "UnpackReshaped", "Unpack"))
         self.concat_ops = set(("Concat", "ConcatV2", "QuantizedConcat", "ConcatTFLite", "PackReshaped", "Pack"))