| AIfES 2 2.0.0

aifes_cnn.h
Go to the documentation of this file.
1
24 #ifndef AIFES_CNN_H
25 #define AIFES_CNN_H
26
27 #ifdef __cplusplus
28 extern "C" {
29 #endif
30
31 // Include the layer base implementations
32 #include "cnn/base/ailayer/ailayer_conv2d.h"
33 #include "cnn/base/ailayer/ailayer_reshape.h"
34 #include "cnn/base/ailayer/ailayer_maxpool2d.h"
35 #include "cnn/base/ailayer/ailayer_batch_normalization.h"
36
37 // ---------------------------- Module default implementations -----------------------
38 // (Fallback functions if no hardware optimized implementation available)
39
40 // Include the math in default implementation
41 #include "cnn/default/aimath/aimath_cnn_f32_default.h"
42
43 // Include the layers in default implementation
44 #include "cnn/default/ailayer/ailayer_conv2d_default.h"
45 #include "cnn/default/ailayer/ailayer_reshape_default.h"
46 #include "cnn/default/ailayer/ailayer_maxpool2d_default.h"
47 #include "cnn/default/ailayer/ailayer_batch_normalization_default.h"
48
49
50 #ifdef __cplusplus
51 } // End extern "C"
52 #endif
53
54 #endif // AIFES_CNN_H
ailayer_batch_normalization.h
Base layer implementation of the Batch Normalization layer.
ailayer_batch_normalization_default.h
Default implementation of the Batch Normalization layer .
ailayer_conv2d.h
Base layer implementation of the Conv2D layer.
ailayer_conv2d_default.h
Default implementation of the Conv2D layer .
ailayer_maxpool2d.h
Base layer implementation of the MaxPool2D layer.
ailayer_maxpool2d_default.h
Default implementation of the MaxPool2D layer .
ailayer_reshape.h
Base layer implementation of the Reshape layer.
ailayer_reshape_default.h
Default implementation of the Reshape layer .
aimath_cnn_f32_default.h
Math functions for F32 data type, CNN-specific implementation.