Feat/operator global average pooling
Compare changes
Files
13@@ -44,10 +44,10 @@ private:
@@ -44,10 +44,10 @@ private:
Adding support for GlobalAveragePooling Operator (needed to implement EfficientNet and MobileNet-v2). No linked gitlab issue.
Contains a fix for GraphView::remove that impaired removeFlatten execution, I added tests for remove flatten & Scheduler at the same time. I refactored a bit GraphView::replace to make the code a bit more readable (more explicit var name and types).
Added an friend ostream operator<< () for node (commented)
Linked MR:
Copyright © Eclipse Foundation, Inc. All Rights Reserved. Privacy Policy | Terms of Use | Copyright Agent