diff --git a/aidge_core/unit_tests/test_forward_dims_constant_shape.py b/aidge_core/unit_tests/test_forward_dims_constant_shape.py index aea0260c8beb94cc880136f9338b16ca400ac25c..fd3fee04b792027e8c33607d1c2156878262d548 100644 --- a/aidge_core/unit_tests/test_forward_dims_constant_shape.py +++ b/aidge_core/unit_tests/test_forward_dims_constant_shape.py @@ -89,7 +89,7 @@ class test_forward_dims_constant_shape(unittest.TestCase): def tearDown(self): pass - def test_attributes_name(self): + def test_constant_shape_folding(self): # Note: Except Div every operator are backend independent self.graph.set_backend("cpu") self.graph.set_datatype(aidge_core.dtype.float32) diff --git a/src/recipes/ShapeFolding.cpp b/src/recipes/ShapeFolding.cpp index ca6ef5087a62262bcf40244c2e1dea2bc15fff81..72d13ab52ca4a9bc84b7b41c6e2349577f207732 100644 --- a/src/recipes/ShapeFolding.cpp +++ b/src/recipes/ShapeFolding.cpp @@ -30,7 +30,7 @@ bool Aidge::constantShapeFolding(std::shared_ptr<GraphView> graph, const std::ve if (not_shape_present) return false; do{ - forwarded = graph->forwardDims(dims, true); + forwarded = graph->forwardDims(dims, false); modified = constantFolding(graph, true); was_modified = true; } while(modified);