ailayer_batch_normalization_default.h

Go to the documentation of this file.

 
#ifndef AILAYER_BATCH_NORM_DEFAULT
#define AILAYER_BATCH_NORM_DEFAULT
 
#include "cnn/base/ailayer/ailayer_batch_normalization.h"
#include "basic/default/aimath/aimath_f32_default.h"
#include "cnn/default/aimath/aimath_cnn_f32_default.h"
 
#define AILAYER_BATCH_NORM_F32_M(momentum, eps, moving_mean, moving_variance, beta, gamma) \
 {{{0,},0,0,0,{0,0,0,0,(float *) beta},{0,0,0,0,(float *) gamma}, \
 {0,0,0,0,(float *) moving_mean},{0,0,0,0,(float *) moving_variance} }, momentum, eps}
#define AILAYER_BATCH_NORM_F32_A(momentum, eps)   {{{0,}}, momentum, eps}
 
 
typedef struct ailayer_batch_norm_f32   ailayer_batch_norm_f32_t;
 
struct ailayer_batch_norm_f32 {
    ailayer_batch_norm_t base; 
    aiscalar_f32_t momentum; 
    aiscalar_f32_t eps; 
};
 
ailayer_t *ailayer_batch_norm_f32_default(ailayer_batch_norm_f32_t *layer, ailayer_t *input_layer);
 
ailayer_t *ailayer_batch_norm_cfirst_f32_default(ailayer_batch_norm_f32_t *layer, ailayer_t *input_layer);
 
ailayer_t *ailayer_batch_norm_chw_f32_default(ailayer_batch_norm_f32_t *layer, ailayer_t *input_layer);
 
ailayer_t *ailayer_batch_norm_cl_f32_default(ailayer_batch_norm_f32_t *layer, ailayer_t *input_layer);
 
 
ailayer_t *ailayer_batch_norm_clast_f32_default(ailayer_batch_norm_f32_t *layer, ailayer_t *input_layer);
 
ailayer_t *ailayer_batch_norm_hwc_f32_default(ailayer_batch_norm_f32_t *layer, ailayer_t *input_layer);
 
ailayer_t *ailayer_batch_norm_lc_f32_default(ailayer_batch_norm_f32_t *layer, ailayer_t *input_layer);
 
 
void ailayer_batch_norm_init_params_f32_default(ailayer_t *self);
 
#endif // AILAYER_BATCH_NORM_DEFAULT