Skip to content
Snippets Groups Projects
Commit 5353b25d authored by Cyril Moineau's avatar Cyril Moineau
Browse files

Update recipes with new type name.

parent d859936e
No related branches found
No related tags found
3 merge requests!279v0.4.0,!253v0.4.0,!163Export refactor
Pipeline #55503 failed
...@@ -24,7 +24,7 @@ ...@@ -24,7 +24,7 @@
#include "aidge/recipes/Recipes.hpp" #include "aidge/recipes/Recipes.hpp"
size_t Aidge::convToMatMul(std::shared_ptr<GraphView> graphView) { size_t Aidge::convToMatMul(std::shared_ptr<GraphView> graphView) {
const auto matches = SinglePassGraphMatching(graphView).match("Conv"); const auto matches = SinglePassGraphMatching(graphView).match("Conv2D");
size_t nbReplaced = 0; size_t nbReplaced = 0;
for (const auto& match : matches) { for (const auto& match : matches) {
......
...@@ -190,9 +190,10 @@ void Aidge::fuseBatchNorm(std::shared_ptr<Aidge::Node> convNode, ...@@ -190,9 +190,10 @@ void Aidge::fuseBatchNorm(std::shared_ptr<Aidge::Node> convNode,
} }
void Aidge::fuseBatchNorm(std::shared_ptr<Aidge::GraphView> graphView) { void Aidge::fuseBatchNorm(std::shared_ptr<Aidge::GraphView> graphView) {
auto matches = SinglePassGraphMatching(graphView).match("(Conv|ConvDepthWise|PaddedConv|PaddedConvDepthWise)->BatchNorm"); auto matches = SinglePassGraphMatching(graphView).match("(Conv2D|ConvDepthWise2D|PaddedConv2D|PaddedConvDepthWise2D)->BatchNorm2D");
for (auto match : matches) { for (auto match : matches) {
fmt::println("Match !");
auto rootNode = match.graph->rootNode(); auto rootNode = match.graph->rootNode();
fuseBatchNorm(rootNode, *rootNode->getChildren().begin()); fuseBatchNorm(rootNode, *rootNode->getChildren().begin());
} }
......
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