Skip to content
Snippets Groups Projects
Commit 5cc3f175 authored by Benjamin Halimi's avatar Benjamin Halimi
Browse files

edit normalizeActivations()

parent b6f593f9
No related branches found
No related tags found
2 merge requests!54Update 0.3.1 -> 0.4.0,!50Enhancement : Quantizer only PTQ
Pipeline #68363 failed
...@@ -818,11 +818,9 @@ void normalizeActivations(std::shared_ptr<GraphView> graphView, std::unordered_m ...@@ -818,11 +818,9 @@ void normalizeActivations(std::shared_ptr<GraphView> graphView, std::unordered_m
// If prevNode is Affine, fix the bias ... // If prevNode is Affine, fix the bias ...
if (isAffine(prevNode)) { if (isAffine(prevNode))
if (nodeHasBias(prevNode)) { if (nodeHasBias(prevNode))
insertScalingBelowProducer(prevNode->getParent(2), 1.0 / prevRatio, graphView); multiplyScalingFactor(prevNode->getParent(2), 1.0 / prevRatio);
}
}
} }
// Merging nodes handling : use a maximum arbritration ... // Merging nodes handling : use a maximum arbritration ...
...@@ -873,7 +871,6 @@ void normalizeActivations(std::shared_ptr<GraphView> graphView, std::unordered_m ...@@ -873,7 +871,6 @@ void normalizeActivations(std::shared_ptr<GraphView> graphView, std::unordered_m
multiplyScalingFactor(prevScalingNode, prevRatio); multiplyScalingFactor(prevScalingNode, prevRatio);
} }
} }
} }
......
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