aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/pass_packing.py
diff options
context:
space:
mode:
authorJacob Bohlin <jacob.bohlin@arm.com>2020-06-29 11:58:50 +0200
committertim.hall <tim.hall@arm.com>2020-07-10 14:20:20 +0000
commit9fbc4913fe5dd0e6090532963f8612449936d994 (patch)
tree01fa3f202116cc4b767077f5e0a3977cf2185559 /ethosu/vela/pass_packing.py
parent0b9ca78dbde3f5c29f368577de1f621845e711a2 (diff)
downloadethos-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/pass_packing.py')
-rw-r--r--ethosu/vela/pass_packing.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py
index fff192d7..c14a70be 100644
--- a/ethosu/vela/pass_packing.py
+++ b/ethosu/vela/pass_packing.py
@@ -100,7 +100,7 @@ npu_post_ops = activation_ops | set(
npu_post_fuse_limited_ops = set(
# Set of post operators that should not be fused with main/elementwise ops
- ("ConcatSliceWrite", "Sigmoid", "Tanh")
+ ("ConcatSliceWrite", "Sigmoid", "Tanh", "Quantize")
)
elem_wise_ops = elem_wise_main_ops | activation_ops | set(("Sigmoid", "Tanh"))