ailayer_dense.h

Go to the documentation of this file.

 
#ifndef AILAYER_DENSE
#define AILAYER_DENSE
 
#include "core/aifes_core.h"
#include "basic/default/aimath/aimath_f32_default.h"
#include "basic/default/aimath/aimath_q7_default.h"
#include "basic/base/aimath/aimath_q31.h"
 
#define DENSE_WEIGHTS_SIZE(INPUTS, OUTPUTS)     ((INPUTS) * (OUTPUTS))
#define DENSE_BIAS_SIZE(OUTPUTS)                (OUTPUTS)
 
#define DENSE_WEIGHTS_SHAPE(INPUTS, OUTPUTS)    {INPUTS, OUTPUTS}
#define DENSE_BIAS_SHAPE(OUTPUTS)               {1, OUTPUTS}
 
typedef struct ailayer_dense    ailayer_dense_t;
 
struct ailayer_dense {
    ailayer_t base; 
    uint32_t neurons; 
 
    aitensor_t weights; 
    aitensor_t bias; 
    uint16_t weights_shape[2]; 
    uint16_t bias_shape[1]; 
    aitensor_t *trainable_params[2]; 
    aitensor_t *gradients[2]; 
    void *optimem[2]; 
 
 
    void (*linear)(const aitensor_t *a, const aitensor_t *b, const aitensor_t *c, aitensor_t *result);
 
    void (*mat_mul_at)(const aitensor_t *a, const aitensor_t *b, aitensor_t *result);
 
    void (*mat_mul_bt)(const aitensor_t *a, const aitensor_t *b, aitensor_t *result);
 
    void (*tensor_add)(const aitensor_t *a, const aitensor_t *b, aitensor_t *result);
 
    void (*sum_channelwise)(const aitensor_t *a, int8_t channel_axis, aitensor_t *result);
 
 
    uint16_t result_shape[2]; 
};
 
extern const aicore_layertype_t *ailayer_dense_type;
 
ailayer_t *ailayer_dense(ailayer_dense_t *layer, ailayer_t *input_layer);
 
void ailayer_dense_forward(ailayer_t *self);
 
void ailayer_dense_backward(ailayer_t *self);
 
void ailayer_dense_calc_result_shape(ailayer_t *self);
 
uint32_t ailayer_dense_sizeof_bwdmem(const ailayer_t *self);
 
uint32_t ailayer_dense_sizeof_paramem(const ailayer_t *self);
 
void ailayer_dense_set_paramem(ailayer_t *self, void *memory_ptr);
 
uint32_t ailayer_dense_sizeof_trainmem(const ailayer_t *self);
 
void ailayer_dense_set_trainmem(ailayer_t *self, void *memory_ptr);
 
#ifdef AIDEBUG_PRINT_MODULE_SPECS
void ailayer_dense_print_specs(const ailayer_t *self);
#endif // AIDEBUG_PRINT_MODULE_SPECS
 
#endif // AILAYER_DENSE