Skip to content
Snippets Groups Projects

Fuse bn

Merged Cyril Moineau requested to merge fuseBN into main
2 files
+ 12
12
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -60,15 +60,15 @@ class test_recipies(unittest.TestCase):
aidge_core.fuse_mul_add(graph_view)
self.assertTrue(len(graph_view.get_nodes()) == len(old_nodes) - 2)
self.assertTrue("MatMul0" not in [i.name for i in graph_view.get_nodes()])
self.assertTrue("Add0" not in [i.name for i in graph_view.get_nodes()])
self.assertTrue("MatMul1" not in [i.name for i in graph_view.get_nodes()])
self.assertTrue("Add1" not in [i.name for i in graph_view.get_nodes()])
self.assertTrue("W0" in [i.name for i in graph_view.get_nodes()])
self.assertTrue("B0" in [i.name for i in graph_view.get_nodes()])
self.assertTrue("W1" in [i.name for i in graph_view.get_nodes()])
self.assertTrue("B1" in [i.name for i in graph_view.get_nodes()])
self.assertTrue("MatMul0" not in [i.name() for i in graph_view.get_nodes()])
self.assertTrue("Add0" not in [i.name() for i in graph_view.get_nodes()])
self.assertTrue("MatMul1" not in [i.name() for i in graph_view.get_nodes()])
self.assertTrue("Add1" not in [i.name() for i in graph_view.get_nodes()])
self.assertTrue("W0" in [i.name() for i in graph_view.get_nodes()])
self.assertTrue("B0" in [i.name() for i in graph_view.get_nodes()])
self.assertTrue("W1" in [i.name() for i in graph_view.get_nodes()])
self.assertTrue("B1" in [i.name() for i in graph_view.get_nodes()])
# TODO : Vérifier que FC bien crée
if __name__ == '__main__':
Loading