nrf_nn_axon.h

Go to the documentation of this file.

/*
 * Copyright (c) 2026 Nordic Semiconductor ASA
 *
 * SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
 */
#ifndef _NRF_NN_AXON_H_
#define _NRF_NN_AXON_H_
 
#include <nrf_edgeai/nn/platform/nrf_nn_platform_types.h>
 
#ifdef __cplusplus
extern "C" {
#endif
 
struct nrf_axon_nn_compiled_model_tag_s;
typedef struct nrf_axon_nn_compiled_model_tag_s nrf_nn_axon_model_t;
 
int8_t nrf_nn_axon_init(const nrf_nn_axon_model_t* p_nn);
 
int8_t nrf_nn_axon_deinit(const nrf_nn_axon_model_t* p_nn);
 
int8_t nrf_nn_axon_init_persistent_vars(const nrf_nn_axon_model_t* p_nn);
 
void nrf_nn_axon_scale_inputs(const nrf_nn_axon_model_t* p_nn,
                              const flt32_t*             p_input_features,
                              size32_t                   input_features_num,
                              int8_t*                    p_scaled_features);
 
int8_t nrf_nn_axon_run_inference(const nrf_nn_axon_model_t* p_nn,
                                 const int8_t*              p_input_features,
                                 int8_t*                    p_output_buffer);
 
void nrf_nn_axon_dequantize_outputs(const nrf_nn_axon_model_t* p_nn,
                                    const int8_t*              p_output_buffer,
                                    uint16_t                   outputs_num,
                                    flt32_t*                   p_dequantized_outputs);
#ifdef __cplusplus
}
#endif
 
#endif /* _NRF_NN_AXON_H_ */