Add support onnx size#3301

laggui merged 2 commits intotracel-ai:maintracel-ai/burn:mainfrom

tye-singwa:onnx-sizetye-singwa/burn:onnx-sizeCopy head branch name to clipboard