Skip to content
Snippets Groups Projects
Commit 4a451782 authored by Matthew  Newson's avatar Matthew Newson
Browse files

Add missing files

parent 9fc76615
No related branches found
No related tags found
1 merge request!33Add and modify operators to run the ConvNeXt onnx model
......@@ -8,4 +8,4 @@
#define {{ name|upper }}_ACTIVATION {{ activation }}
#define {{ name|upper }}_EPSILON {{ epsilon }}
#endif /* {{ name|upper }}_LAYER_H */
#endif /* {{ name|upper }}_LAYER_H */
\ No newline at end of file
......@@ -7,7 +7,6 @@
{% include "./_meminfo.jinja" %}
{# For layer configuration -#}
<<<<<<< HEAD
{% include "./_def_io.jinja" %}
{% include "./_meminfo.jinja" %}
#define {{ name|upper }}_B {{ in_dims[0][0]}}
......@@ -24,11 +23,6 @@ int {{name|upper}}_OUTPUT_DIMS[] = { {{ out_dims[0]|join(", ") }} };
int {{name|upper}}_INPUT_A_DIMS[] = { {{ in_dims[0]|join(", ") }} };
int {{name|upper}}_INPUT_B_DIMS[] = { {{ in_dims[1]|join(", ") }} };
=======
#define {{ name|upper }}_M {{ in_dims[0][0] }}
#define {{ name|upper }}_K {{ in_dims[0][1] }}
#define {{ name|upper }}_N {{ in_dims[1][1] }}
>>>>>>> origin/dev
#define {{ name|upper }}_ACTIVATION {{ activation }}
{#- Calculate sizes #}
......
......@@ -6,4 +6,4 @@ batchnorm_forward<{{ out_name[0]|upper }}_NB_OUTPUTS,
{{name|upper}}_ACTIVATION>
({{in_name[0]}}, {{out_name[0]}}, {{in_name[1]}}, {{in_name[2]}}, {{in_name[3]}}, {{in_name[4]}}, {{name|upper}}_EPSILON);
{% include "./_save_outputs.jinja" %}
{% endfilter %}
{% endfilter %}
\ No newline at end of file
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