Skip to content
Snippets Groups Projects
Commit c72abf5b authored by Cyril Moineau's avatar Cyril Moineau
Browse files

Move files to core.

parent e2f3a04b
No related branches found
No related tags found
3 merge requests!27v0.2.0,!22v0.4.0,!15Export refactor
#include <iostream>
#include "dnn.hpp"
#include "inputs.h"
int main()
{
// Example for MNIST dataset
// Feel free to change this file for your own projects
const unsigned int nb_classes = 10;
float results[nb_classes];
model_forward(inputs, results);
for (unsigned int i = 0; i < nb_classes; ++i)
{
std::cout << i << ": " << results[i] << std::endl;
}
return 0;
}
\ No newline at end of file
{#- For name header -#}
#ifndef MEM_INFO_H
#define MEM_INFO_H
#define MEMORY_SIZE {{ mem_size }}
{% for i in range(mem_info|length) -%}
{%- set layer_name = mem_info[i][0] %}
/* {{layer_name}} memory */
{% for j in range(1, mem_info[i]|length) %}
#define {{ layer_name|upper }}_{{ mem_info_legends[j]|upper }} {{ mem_info[i][j] }}
{%- endfor %}
{% endfor %}
#endif /* MEM_INFO_H */
\ No newline at end of file
{#- For name header -#}
#ifndef DNN_HPP
#define DNN_HPP
{#- For libraries #}
{% for lib in libraries %}
#include <{{ lib }}>
{%- endfor %}
{% for func in functions %}
{{ func }}
{% endfor %}
#endif /* DNN_HPP */
\ 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