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

edit quantizeNormalizedNetwork() (MatMul support)

parent 3b9e29ce
No related branches found
No related tags found
3 merge requests!54Update 0.3.1 -> 0.4.0,!49Forked from add_matmul (merged automatically),!45Add support for the MatMul operator
Pipeline #67837 passed
......@@ -1046,6 +1046,10 @@ void quantizeNormalizedNetwork(std::shared_ptr<GraphView> graphView, std::uint8_
std::shared_ptr<Node> scalingNode = getUniqueChild(node); // TODO : assert if scalingNode is a Scaling ...
// TODO : double check this ...
if (node->type() == "MatMul")
rescaling /= inputIsUnsigned ? unsignedMax : signedMax;
multiplyScalingFactor(scalingNode, rescaling) ;
}
......
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