aimath_q31.h

Go to the documentation of this file.

 
#ifndef AIMATH_Q31
#define AIMATH_Q31
 
#include "core/aifes_math.h"
#include "basic/base/aimath/aimath_basic.h"
 
#define AITENSOR_2D_Q31(shape, params, data)    {aiq31, 2, shape, params, data}
 
 
#define Q31_TO_FLOAT(Q, S, Z)       ((float) ((int64_t)(Q) - (int64_t)(Z)) / (float) ((int64_t) 1ULL << (S)))
 
#define FLOAT_TO_Q31(F, S, Z)       ((int32_t) ((int64_t)((F) * (1ULL << (S)) + ((F) >= 0 ? 0.5f : -0.5f)) + (int64_t)(Z)))
 
#define AISCALAR_Q31(F, S, Z)       {FLOAT_TO_Q31(F, S, Z), S, Z}
 
typedef struct aimath_q31_params    aimath_q31_params_t;
typedef struct aiscalar_q31         aiscalar_q31_t;
 
struct aimath_q31_params {
    uint16_t shift; 
    int32_t zero_point; 
};
 
struct aiscalar_q31 {
    int32_t value; 
    uint16_t shift; 
    int32_t zero_point; 
};
 
void aimath_q31_print_aitensor(const aitensor_t *tensor);
 
void aimath_q31_print_aiscalar(const void *scalar);
 
void aimath_q31_calc_q_params_from_f32(float min_value, float max_value, aimath_q31_params_t *q_params);
 
void aimath_q31_quantize_tensor_from_f32(const aitensor_t *tensor_f32, aitensor_t *tensor_q31);
 
extern const aimath_dtype_t *aiq31;
 
#endif // AIMATH_Q31