spec/ml_quantize.h

Go to the documentation of this file.

/* SPDX-License-Identifier: BSD-3-Clause
 * Copyright (c) 2023 Nokia
 * Copyright (c) 2024 Marvell
 */
 
#ifndef ODP_API_SPEC_ML_QUANTIZE_H_
#define ODP_API_SPEC_ML_QUANTIZE_H_
#include <odp/visibility_begin.h>
 
#ifdef __cplusplus
extern "C" {
#endif
 
#include <odp/api/std_types.h>
 
void odp_ml_fp32_to_uint8(uint8_t *dst_u8, const float *src_fp32, uint32_t num,
              float scale, uint8_t zerop);
 
void odp_ml_fp32_from_uint8(float *dst_fp32, const uint8_t *src_u8, uint32_t num,
                float scale, uint8_t zerop);
 
void odp_ml_fp32_to_int8(int8_t *dst_i8, const float *src_fp32, uint32_t num, float scale,
             int8_t zerop);
 
void odp_ml_fp32_from_int8(float *dst_fp32, const int8_t *src_i8, uint32_t num, float scale,
               int8_t zerop);
 
void odp_ml_fp32_to_uint16(uint16_t *dst_u16, const float *src_fp32, uint32_t num, float scale,
               uint16_t zerop);
 
void odp_ml_fp32_from_uint16(float *dst_fp32, const uint16_t *src_u16, uint32_t num, float scale,
                 uint16_t zerop);
 
void odp_ml_fp32_to_int16(int16_t *dst_i16, const float *src_fp32, uint32_t num, float scale,
              int16_t zerop);
 
void odp_ml_fp32_from_int16(float *dst_fp32, const int16_t *src_i16, uint32_t num, float scale,
                int16_t zerop);
 
void odp_ml_fp32_to_fp16(uint16_t *dst_fp16, const float *src_fp32, uint32_t num);
 
void odp_ml_fp32_from_fp16(float *dst_fp32, const uint16_t *src_fp16, uint32_t num);
 
#ifdef __cplusplus
}
#endif
 
#include <odp/visibility_end.h>
#endif