ailayer_conv2d_default.h

Go to the documentation of this file.

 
#ifndef AILAYER_CONV2D_DEFAULT
#define AILAYER_CONV2D_DEFAULT
 
#include "cnn/base/ailayer/ailayer_conv2d.h"
 
#include "cnn/default/aimath/aimath_cnn_f32_default.h"
#include "basic/default/aimath/aimath_f32_default.h"
 
#define HW(h, w)        {h, w}
 
#define AILAYER_CONV2D_F32_M(filters, kernel_size, stride, dilation, padding, weights, bias) \
            {{0,},filters,kernel_size,stride,dilation,padding,0,{0,0,0,0,(float *) weights},{0,0,0,0,(float *) bias}}
#define AILAYER_CONV2D_F32_A(filters, kernel_size, stride, dilation, padding) \
            {{0,},filters,kernel_size,stride,dilation,padding,0,{0,0,0,0,0},{0,0,0,0,0}}
 
typedef struct ailayer_conv2d   ailayer_conv2d_f32_t;
 
ailayer_t *ailayer_conv2d_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer);
 
 
ailayer_t *ailayer_conv2d_cfirst_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer);
 
ailayer_t *ailayer_conv2d_chw_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer);
 
 
ailayer_t *ailayer_conv2d_clast_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer);
 
ailayer_t *ailayer_conv2d_hwc_f32_default(ailayer_conv2d_f32_t *layer, ailayer_t *input_layer);
 
void ailayer_conv2d_init_params_f32_default(ailayer_t *self);
 
#endif // AILAYER_CONV2D_DEFAULT