Baremetal-NN
Baremetal-NN API documentation
Loading...
Searching...
No Matches
nn.h
Go to the documentation of this file.
1#ifndef __NN_H
2#define __NN_H
3
4#include <assert.h>
5
6#include "float16.h"
7#include "tensor.h"
8#include "functional/tensor_creation.h"
9#include "functional/print.h"
10#include "functional/abs.h"
11#include "functional/add.h"
12#include "functional/batch_norm2d.h"
13#include "functional/clip.h"
14#include "functional/conv2d.h"
15#include "functional/copy.h"
16#include "functional/div.h"
17#include "functional/elu.h"
18#include "functional/fill.h"
19#include "functional/interpolate.h"
20#include "functional/layer_norm.h"
21#include "functional/linear.h"
22#include "functional/matmul.h"
23#include "functional/norm.h"
24#include "functional/max.h"
25#include "functional/mm.h"
26#include "functional/maximum.h"
27#include "functional/min.h"
28#include "functional/minimum.h"
29#include "functional/mul.h"
30#include "functional/mv.h"
31#include "functional/neg.h"
32#include "functional/relu.h"
33#include "functional/relu6.h"
34#include "functional/rms_norm.h"
35#include "functional/softmax.h"
36#include "functional/silu.h"
37#include "functional/sub.h"
38#include "functional/sum.h"
39
40
41// http://elm-chan.org/junk/32bit/binclude.html
42#define INCLUDE_FILE(section, filename, symbol) asm (\
43 ".section "#section"\n" /* Change section */\
44 ".balign 4\n" /* Word alignment */\
45 ".global "#symbol"_start\n" /* Export the object start address */\
46 ".global "#symbol"_data\n" /* Export the object address */\
47 #symbol"_start:\n" /* Define the object start address label */\
48 #symbol"_data:\n" /* Define the object label */\
49 ".incbin \""filename"\"\n" /* Import the file */\
50 ".global "#symbol"_end\n" /* Export the object end address */\
51 #symbol"_end:\n" /* Define the object end address label */\
52 ".balign 4\n" /* Word alignment */\
53 ".section \".text\"\n") /* Restore section */
54
55
56
57void NN_assert(int condition, char *message) {
58 if (!condition) {
59 printf("Assertion failed: ");
60 printf("%s\n", message);
61 exit(1);
62 }
63}
64
65
66
67#endif // __NN_H
batch_norm2d.h
rms_norm.h
layer_norm.h
NN_assert
void NN_assert(int condition, char *message)
Definition: nn.h:57
tensor_creation.h