Skip to content
Snippets Groups Projects
Commit a848e358 authored by Mickael GUIBERT's avatar Mickael GUIBERT Committed by Axel Farrugia
Browse files

[Feat] Add jinja file for forward build

parent 4bfbd307
No related branches found
No related tags found
No related merge requests found
{#- For name header -#}
#ifndef {{ name|upper }}_LAYER_H
#define {{ name|upper }}_LAYER_H
{# For layer configuration -#}
{% include "./_def_io.jinja" %}
{% include "./_meminfo.jinja" %}
#define {{ name|upper }}_NB_DATA {{ in_chan[0] * in_height[0] * in_width[0] }}
// Activation
#define {{ name|upper }}_ACTIVATION {{ activation }}
{% include "./_rescaling.jinja" %}
#endif /* {{ name|upper }}_LAYER_H */
{% filter indent(width=4, first=False) %}
{% include "./_mem_offset.jinja" %}
rescaling_forward<{{name|upper}}_NB_DATA,
{{name|upper}}_ACTIVATION>
({{in_name[0]}},
{{out_name[0]}},
{{name|upper}}_RESCALING);
{% include "./_save_outputs.jinja" %}
{% endfilter %}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment