From 28474e96b4216349dfa6fd0fd31053c5dd082b72 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 24 Mar 2020 13:18:28 +0000
Subject: [PATCH] [Transform] call InferDataType after
AbsorbTransposeIntoMultiT
---
src/finn/transformation/streamline/absorb.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/src/finn/transformation/streamline/absorb.py b/src/finn/transformation/streamline/absorb.py
index 5795d9d71..0d709297a 100644
--- a/src/finn/transformation/streamline/absorb.py
+++ b/src/finn/transformation/streamline/absorb.py
@@ -33,6 +33,7 @@ from finn.core.datatype import DataType
from finn.transformation import Transformation
from finn.util.basic import get_by_name
from finn.custom_op.registry import getCustomOp
+from finn.transformation.infer_datatypes import InferDataTypes
class AbsorbAddIntoMultiThreshold(Transformation):
@@ -278,4 +279,6 @@ class AbsorbTransposeIntoMultiThreshold(Transformation):
model.set_tensor_shape(mt_cand.output[0], mt_ishape)
graph.node.remove(n)
graph_modified = True
+ if graph_modified:
+ model = model.transform(InferDataTypes())
return (model, graph_modified)
--
GitLab