diff options
author | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-06-29 11:58:50 +0200 |
---|---|---|
committer | tim.hall <tim.hall@arm.com> | 2020-07-10 14:20:20 +0000 |
commit | 9fbc4913fe5dd0e6090532963f8612449936d994 (patch) | |
tree | 01fa3f202116cc4b767077f5e0a3977cf2185559 /ethosu/vela/supported_operators.py | |
parent | 0b9ca78dbde3f5c29f368577de1f621845e711a2 (diff) | |
download | ethos-u-vela-9fbc4913fe5dd0e6090532963f8612449936d994.tar.gz |
MLBEDSW-1497: Add Quantize operator support
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: Iaf4d7ab9c32b0d783072c5f131a61bfebe77cc16
Diffstat (limited to 'ethosu/vela/supported_operators.py')
-rw-r--r-- | ethosu/vela/supported_operators.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py index 55e718e9..3bf46a9d 100644 --- a/ethosu/vela/supported_operators.py +++ b/ethosu/vela/supported_operators.py @@ -66,6 +66,8 @@ class SupportedOperators: | 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")) |