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

[Streamline] remove identity ops as part of streamlining

parent 59b06427
No related branches found
No related tags found
No related merge requests found
......@@ -59,6 +59,7 @@ from finn.transformation.streamline.reorder import (
from finn.transformation.streamline.round_thresholds import RoundAndClipThresholds
from finn.transformation.streamline.sign_to_thres import ConvertSignToThres
from finn.transformation.batchnorm_to_affine import BatchNormToAffine
from finn.transformation.streamline.remove import RemoveIdentityOps
class Streamline(Transformation):
......@@ -87,6 +88,7 @@ class Streamline(Transformation):
]
for trn in streamline_transformations:
model = model.transform(trn)
model = model.transform(RemoveIdentityOps())
model = model.transform(GiveUniqueNodeNames())
model = model.transform(GiveReadableTensorNames())
model = model.transform(InferDataTypes())
......
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