ra8_npu_quant.h

Go to the documentation of this file.

 
#pragma once
 
#include <stddef.h>
#include <stdint.h>
 
#include "ra8_err.h"
 
#ifdef __cplusplus
extern "C" {
#endif

typedef enum : int32_t {
  k_ra8_npu_quant_i8_min = -128, 
  k_ra8_npu_quant_i8_max = 127,  
  k_ra8_npu_quant_u8_min = 0,    
  k_ra8_npu_quant_u8_max = 255,  
} ra8_npu_quant_range_t;

[[nodiscard]] ra8_err_t
ra8_npu_quantize_i8(const float* in, int8_t* out, size_t count, float scale, int32_t zero_point);

[[nodiscard]] ra8_err_t
ra8_npu_dequantize_i8(const int8_t* in, float* out, size_t count, float scale, int32_t zero_point);

[[nodiscard]] ra8_err_t
ra8_npu_quantize_u8(const float* in, uint8_t* out, size_t count, float scale, int32_t zero_point);

[[nodiscard]] ra8_err_t
ra8_npu_dequantize_u8(const uint8_t* in, float* out, size_t count, float scale, int32_t zero_point);
 
#ifdef __cplusplus
}
#endif