aiopti_sgd_default.h

Go to the documentation of this file.

 
#ifndef AIOPTI_SGD_DEFAULT
#define AIOPTI_SGD_DEFAULT
 
#include "basic/base/aiopti/aiopti_sgd.h"
 
#include "basic/default/aimath/aimath_f32_default.h"
#include "basic/default/aimath/aimath_q31_default.h"
#include "basic/default/aimath/aimath_q7_default.h"
 
#define AIOPTI_SGD_F32(learning_rate) {{{0,},},learning_rate, 0.0f}
#define AIOPTI_SGD_WITH_MOMENTUM_F32(learning_rate, momentum) {{{0,},},learning_rate, momentum}
#define AIOPTI_SGD_Q31(learning_rate) {{{0,},},learning_rate, {0,0,0}}
#define AIOPTI_SGD_WITH_MOMENTUM_Q31(learning_rate, momentum) {{{0,},},learning_rate, momentum}
 
typedef struct aiopti_sgd_f32   aiopti_sgd_f32_t; 
typedef struct aiopti_sgd_q31   aiopti_sgd_q31_t; 
typedef struct aiopti_sgd_q7    aiopti_sgd_q7_t; 
struct aiopti_sgd_f32 {
    aiopti_sgd_t base; 
    aiscalar_f32_t learning_rate; 
    aiscalar_f32_t momentum; 
};
 
struct aiopti_sgd_q31 {
    aiopti_sgd_t base; 
    aiscalar_q31_t learning_rate; 
    aiscalar_q31_t momentum; 
};
 
aiopti_t *aiopti_sgd_f32_default(aiopti_sgd_f32_t *opti);
 
aiopti_t *aiopti_sgd_q31_default(aiopti_sgd_q31_t *opti);
 
#endif // AIOPTI_SGD_DEFAULT