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

handle the MatMuls in prepareNetwork()

parent 718f6a1f
Branches
Tags
2 merge requests!54Update 0.3.1 -> 0.4.0,!45Add support for the MatMul operator
Pipeline #68061 passed
......@@ -122,6 +122,12 @@ void prepareNetwork(std::shared_ptr<GraphView> graphView)
std::vector<std::shared_ptr<Node>> nodeVector = retrieveNodeVector(graphView);
// handle the MatMuls
reorderMatMulInputs(graphView);
matMulToFC(graphView);
// tag the weighted nodes
for (std::shared_ptr<Node> node : nodeVector)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment