| AIfES 2 2.0.0

ailayer_conv2d_default.h
Go to the documentation of this file.
1
24 #ifndef AILAYER_CONV2D_DEFAULT
25 #define AILAYER_CONV2D_DEFAULT
26
27 #include "cnn/base/ailayer/ailayer_conv2d.h"
28
29 #include "cnn/default/aimath/aimath_cnn_f32_default.h"
30 #include "basic/default/aimath/aimath_f32_default.h"
31
32 #define HW(h, w) {h, w}
33
34 #define AILAYER_CONV2D_F32_M(filters, kernel_size, stride, dilation, padding, weights, bias) \
35 {{0,},filters,kernel_size,stride,dilation,padding,0,{0,0,0,0,(float *) weights},{0,0,0,0,(float *) bias}}
36 #define AILAYER_CONV2D_F32_A(filters, kernel_size, stride, dilation, padding) \
37 {{0,},filters,kernel_size,stride,dilation,padding,0,{0,0,0,0,0},{0,0,0,0,0}}
38
39 typedef struct ailayer_conv2d ailayer_conv2d_f32_t;
40
137 ailayer_t *ailayer_conv2d_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer);
138
139
148 ailayer_t *ailayer_conv2d_cfirst_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer);
149
158 ailayer_t *ailayer_conv2d_chw_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer);
159
160
169 ailayer_t *ailayer_conv2d_clast_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer);
170
179 ailayer_t *ailayer_conv2d_hwc_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer);
180
194 void ailayer_conv2d_init_params_f32_default(ailayer_t *self);
195
196 #endif // AILAYER_CONV2D_DEFAULT
197
198
ailayer_conv2d.h
Base layer implementation of the Conv2D layer.
ailayer_conv2d_chw_f32_default
ailayer_t * ailayer_conv2d_chw_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer)
Initializes and connect a Conv2D layer (channels first) with the F32 default implementation.
ailayer_conv2d_clast_f32_default
ailayer_t * ailayer_conv2d_clast_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer)
Initializes and connect a Conv2D layer (channels last) with the F32 default implementation.
ailayer_conv2d_init_params_f32_default
void ailayer_conv2d_init_params_f32_default(ailayer_t *self)
F32 default implementation of the ailayer.init_params function for the Conv2D layer
ailayer_conv2d_f32_default
ailayer_t * ailayer_conv2d_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer)
Initializes and connect a Conv2D layer with the F32 default implementation.
ailayer_conv2d_hwc_f32_default
ailayer_t * ailayer_conv2d_hwc_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer)
Initializes and connect a Conv2D layer (channels last) with the F32 default implementation.
ailayer_conv2d_cfirst_f32_default
ailayer_t * ailayer_conv2d_cfirst_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer)
Initializes and connect a Conv2D layer (channels first) with the F32 default implementation.
aimath_cnn_f32_default.h
Math functions for F32 data type, CNN-specific implementation.
aimath_f32_default.h
Math functions for F32 data type, default implementation.
ailayer_conv2d
General Conv2D layer structure.
Definition: ailayer_conv2d.h:57
ailayer
AIfES layer interface.
Definition: aifes_core.h:252