Adding some export features
Context
This merge request only concerns Exports.
It aims to add back some control from the export itself as it adds the possibility to specify :
- The memory section in which the feature maps are stored;
- The main file template;
- Whether the network contains labels for the results verification within the main file.
Modified files
Memory Section
-
forward.jinja
: Add the section as attribute while creating the mem buffer. -
scheduler_exort.py
: Add themem_section
as attribute to be read by the template. -
export_registry.py
: Add themem_section
as class variable to be changed by the export.
Main file generation
-
export_registry.py
: Add the template file path as class variable to be specified by the export. -
scheduler_export.py
: Add the template file path as argument to generate the main file. Note that the aidge template file will still be used by default.
Labels
-
scheduler_export.py
: Add an optional argumentlabels
to thescheduler_export()
function which is then passed as attribute while generating the main file.
Edited by Axel Farrugia