aimath_q7.h

Go to the documentation of this file.

 
#ifndef AIMATH_Q7
#define AIMATH_Q7
 
#include "core/aifes_math.h"
#include "basic/base/aimath/aimath_basic.h"
 
#define AITENSOR_2D_Q7(shape, params, data)    {aiq7, 2, shape, params, data}
 
#define Q7_TO_FLOAT(Q, S, Z)        ((float) ((int16_t)(Q) - (int16_t)(Z)) / (float) (1 << (S)))
 
#define FLOAT_TO_Q7(F, S, Z)        ((int8_t) ((int16_t)((F) * (1 << (S)) + ((F) >= 0 ? 0.5f : -0.5f)) + (int16_t)(Z)))
 
#define AISCALAR_Q7(F, S, Z)       {FLOAT_TO_Q7(F, S, Z), S, Z}
 
typedef struct aimath_q7_params aimath_q7_params_t;
typedef struct aiscalar_q7  aiscalar_q7_t;
 
struct aimath_q7_params {
    uint16_t shift;  
    int8_t zero_point; 
};
 
struct aiscalar_q7 {
    int8_t value; 
    uint16_t shift; 
    int8_t zero_point; 
};
 
void aimath_q7_print_aitensor(const aitensor_t *tensor);
 
void aimath_q7_print_aiscalar(const void *scalar);
 
void aimath_q7_calc_q_params_from_f32(float min_value, float max_value, aimath_q7_params_t *q_params);
 
void aimath_q7_quantize_tensor_from_f32(const aitensor_t *tensor_f32, aitensor_t *tensor_q7);
 
extern const aimath_dtype_t *aiq7;
 
#endif // AIMATH_Q7