Skip to content
Snippets Groups Projects
Commit 96bda2a3 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Streamline] add Absorb1BitMulIntoConv into Streamline

parent 6484f25a
No related branches found
No related tags found
No related merge requests found
...@@ -39,6 +39,7 @@ from finn.transformation.streamline.absorb import ( ...@@ -39,6 +39,7 @@ from finn.transformation.streamline.absorb import (
AbsorbMulIntoMultiThreshold, AbsorbMulIntoMultiThreshold,
FactorOutMulSignMagnitude, FactorOutMulSignMagnitude,
Absorb1BitMulIntoMatMul, Absorb1BitMulIntoMatMul,
Absorb1BitMulIntoConv,
) )
from finn.transformation.streamline.collapse_repeated import ( from finn.transformation.streamline.collapse_repeated import (
...@@ -75,6 +76,7 @@ class Streamline(Transformation): ...@@ -75,6 +76,7 @@ class Streamline(Transformation):
FactorOutMulSignMagnitude(), FactorOutMulSignMagnitude(),
AbsorbMulIntoMultiThreshold(), AbsorbMulIntoMultiThreshold(),
Absorb1BitMulIntoMatMul(), Absorb1BitMulIntoMatMul(),
Absorb1BitMulIntoConv(),
RoundAndClipThresholds(), RoundAndClipThresholds(),
] ]
for trn in streamline_transformations: for trn in streamline_transformations:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment