Skip to content
Snippets Groups Projects
Commit 9cddbffc authored by Maxence Naud's avatar Maxence Naud
Browse files

[Add] First horizontal tiling declaration

parent 03bf2195
No related branches found
No related tags found
No related merge requests found
...@@ -47,7 +47,7 @@ void removeFlatten(std::set<std::shared_ptr<Node>> nodes); ...@@ -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. * @param graphView Graph view to use graph matching on, in order to apply transfomrations.
*/ */
void removeFlatten(std::shared_ptr<GraphView> graphView); void removeFlatten(std::shared_ptr<GraphView> graphView);
// FUSE BN + FC || CONV -> FC || CONV // FUSE BN + FC || CONV -> FC || CONV
/** /**
...@@ -65,6 +65,9 @@ void fuseBatchNorm(std::set<std::shared_ptr<Node>> nodes); ...@@ -65,6 +65,9 @@ void fuseBatchNorm(std::set<std::shared_ptr<Node>> nodes);
*/ */
void fuseBatchNorm(std::shared_ptr<GraphView> graphView); 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_ */ #endif /* AIDGE_CORE_UTILS_RECIPIES_H_ */
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