diff --git a/aidge_export_cpp/templates/configuration/_def_io.jinja b/aidge_export_cpp/templates/configuration/_def_io.jinja index e8330d9fa1b156530ed1e3f460f669c2379e472a..66756cf8f501035f7222272f9c410908f499f06f 100644 --- a/aidge_export_cpp/templates/configuration/_def_io.jinja +++ b/aidge_export_cpp/templates/configuration/_def_io.jinja @@ -1,10 +1,12 @@ {# NOTE: Suppose input is first #} - +// INPUT CONF {% for inidx in range(nb_in) -%} #define {{ in_name[inidx]|upper }}_NB_CHANNELS {{ in_chan[inidx] }} #define {{ in_name[inidx]|upper }}_IN_HEIGHT {{ in_height[inidx] }} #define {{ in_name[inidx]|upper }}_IN_WIDTH {{ in_width[inidx] }} {% endfor %} + +// OUTPUT CONF {% for outidx in range(nb_out) -%} #define {{ out_name[outidx]|upper }}_NB_OUTPUTS {{ out_chan[outidx] }} #define {{ out_name[outidx]|upper }}_OUT_HEIGHT {{ out_height[outidx] }} diff --git a/aidge_export_cpp/templates/configuration/_meminfo.jinja b/aidge_export_cpp/templates/configuration/_meminfo.jinja index 9fa522847b495ab81df284dc66a79d736949ea7d..15d15425b4330f68b4a97c31e9cf7a1076cc93e8 100644 --- a/aidge_export_cpp/templates/configuration/_meminfo.jinja +++ b/aidge_export_cpp/templates/configuration/_meminfo.jinja @@ -1,3 +1,4 @@ +// MEMINFO CONF {% for outidx in range(nb_out) -%} #define {{ out_name[outidx]|upper }}_SIZE {{ mem_info_size[outidx]}} #define {{ out_name[outidx]|upper }}_OFFSET {{ mem_info_offset[outidx]}}