From 2fdaf1dc35c14d8984229550827cf6459ecf429a Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Mon, 14 Sep 2020 20:38:05 +0200
Subject: [PATCH] [Transform] support Div in MoveScalarLinearPastInvariants
---
src/finn/transformation/streamline/reorder.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/finn/transformation/streamline/reorder.py b/src/finn/transformation/streamline/reorder.py
index b47f269dd..f4c1dc130 100644
--- a/src/finn/transformation/streamline/reorder.py
+++ b/src/finn/transformation/streamline/reorder.py
@@ -533,7 +533,7 @@ class MoveScalarLinearPastInvariants(Transformation):
if prod0 is None:
continue
- if prod0.op_type == "Mul" or prod0.op_type == "Add":
+ if prod0.op_type in ["Mul", "Add", "Div"]:
# check if second input of producer is an initializer
init0 = model.get_initializer(prod0.input[1])
# if either initializer is None, skip
--
GitLab