set_dataformat() takes a really long time for large models
Context
The set_dataformat()
step, applied on the entire model during the export process, may transpose the output tensors of each node.
This is typically the case for the export cpp, as all current implementations require NHWC format whereas Aidge native format is NCHW.
Issue
This transpose process can take some time, particularly for large models (such as VGG16).
Maybe there's a way to speed up the process ?