From 9cddbffc8c2b89db37eaba988cdfc16cfff36ae2 Mon Sep 17 00:00:00 2001 From: NAUD Maxence <maxence.naud@cea.fr> Date: Wed, 4 Oct 2023 14:38:04 +0000 Subject: [PATCH] [Add] First horizontal tiling declaration --- include/aidge/recipies/Recipies.hpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/include/aidge/recipies/Recipies.hpp b/include/aidge/recipies/Recipies.hpp index 894e56fae..7d2539f7a 100644 --- a/include/aidge/recipies/Recipies.hpp +++ b/include/aidge/recipies/Recipies.hpp @@ -47,7 +47,7 @@ void removeFlatten(std::set<std::shared_ptr<Node>> nodes); * @param graphView Graph view to use graph matching on, in order to apply transfomrations. */ void removeFlatten(std::shared_ptr<GraphView> graphView); - + // FUSE BN + FC || CONV -> FC || CONV /** @@ -65,6 +65,9 @@ void fuseBatchNorm(std::set<std::shared_ptr<Node>> nodes); */ void fuseBatchNorm(std::shared_ptr<GraphView> graphView); +std::set<std::shared_ptr<Node>> horizontalTiling(std::shared_ptr<Node> node); +std::set<std::shared_ptr<Node>> horizontalTiling(std::set<std::shared_ptr<Node>> setOfNodes); + } #endif /* AIDGE_CORE_UTILS_RECIPIES_H_ */ -- GitLab