Come up with a clear way of parametrising the ONNX export
This would allow for instance, according to what we need
- Convert the internal nodes from INT64 to INT32
- Convert a model to
fp16
- Import the
scatter_add
operation as a custom operation
This should result in a different name for the exported ONNX file.