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);
* @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_ */
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