Cyril Moineau
--- a/src/operator/Shape.cpp
+++ b/src/operator/Shape.cpp
        AIDGE_ASSERT(roi> 1, "Invalid ROI for Shape");

        mOutputs[0]->resize({roi});
+        if (!mOutputs[0]->getImpl()){
+            Log::debug("Shape::forwardDims, no implementation set for output, defaulting to CPU.");
+            mOutputs[0]->setBackend("cpu");
--- a/unit_tests/operator/Test_ShapeImpl.cpp
+++ b/unit_tests/operator/Test_ShapeImpl.cpp
            {1, 2, 3, 5}
        });

-        std::shared_ptr<Node> myShape = Shape();
-        auto op = std::static_pointer_cast<OperatorTensor>(myShape -> getOperator());
-        op->associateInput(0,input);
+        std::shared_ptr<Shape_Op> op = std::make_shared<Shape_Op>();
+        op->associateInput(0, input);
        op->setDataType(DataType::Int32);
        op->setBackend("cpu");
-        myShape->forward();
+        op->forward();