Kernel crashes with forward_dims method - Which argument in the set_compute_output_dims method ?

I am working on aidge with a YOLOv2 model. There is one Generic Operator : Reshape

My goal here is to implement this Reshape operator by following the tutorial about implementations.

Here is my code : aidge_yolov2.py.

This is the output log : log

Sorry, the onnx model file is too heavy, I can't share it :/

[UPDATED] : see the comments below