nrf_nn_neuton.h

Go to the documentation of this file.

/*
 * Copyright (c) 2026 Nordic Semiconductor ASA
 *
 * SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
 */
#ifndef _NRF_NN_NEUTON_H_
#define _NRF_NN_NEUTON_H_
 
#include <nrf_edgeai/nn/platform/nrf_nn_platform_types.h>
 
#ifdef __cplusplus
extern "C" {
#endif
 
typedef struct nrf_nn_neuton_model_meta_s
{
    const uint16_t* p_neuron_internal_links_num; 
    const uint16_t* p_neuron_external_links_num; 
    const uint16_t* p_output_neurons_indices;    
    const uint16_t* p_neuron_links;              
    const uint8_t*  p_neuron_act_type_mask;      
    const uint16_t  outputs_num;                 
    const uint16_t  neurons_num;                 
    const uint32_t  weights_num;                 
} nrf_nn_neuton_model_meta_t;
 
typedef struct nrf_nn_neuton_model_params_q8_s
{
    const int8_t*  p_weights;     
    const uint8_t* p_act_weights; 
    uint8_t*       p_neurons;     
} nrf_nn_neuton_model_params_q8_t;
 
typedef struct nrf_nn_neuton_model_params_q16_s
{
    const int16_t*  p_weights;     
    const uint16_t* p_act_weights; 
    uint16_t*       p_neurons;     
} nrf_nn_neuton_model_params_q16_t;
 
typedef struct nrf_nn_neuton_model_params_f32_s
{
    const flt32_t* p_weights;     
    const flt32_t* p_act_weights; 
    flt32_t*       p_neurons;     
} nrf_nn_neuton_model_params_f32_t;
 
typedef struct nrf_nn_neuton_model_q8_s
{
    nrf_nn_neuton_model_meta_t      meta;   
    nrf_nn_neuton_model_params_q8_t params; 
} nrf_nn_neuton_model_q8_t;
 
typedef struct nrf_nn_neuton_model_q16_s
{
    nrf_nn_neuton_model_meta_t       meta;   
    nrf_nn_neuton_model_params_q16_t params; 
} nrf_nn_neuton_model_q16_t;
 
typedef struct nrf_nn_neuton_model_f32_s
{
    nrf_nn_neuton_model_meta_t       meta;   
    nrf_nn_neuton_model_params_f32_t params; 
} nrf_nn_neuton_model_f32_t;
 
void nrf_nn_neuton_run_inference_q8(nrf_nn_neuton_model_q8_t* p_model,
                                    const uint8_t*            p_inputs,
                                    const uint16_t            inputs_num);
 
void nrf_nn_neuton_run_inference_q16(nrf_nn_neuton_model_q16_t* p_model,
                                     const uint16_t*            p_inputs,
                                     const uint16_t             inputs_num);
 
void nrf_nn_neuton_run_inference_f32(nrf_nn_neuton_model_f32_t* p_model,
                                     const flt32_t*             p_inputs,
                                     const uint16_t             inputs_num);
 
void nrf_nn_neuton_run_inference_raw_f32(flt32_t*        p_neurons,
                                         const uint32_t  neurons_num,
                                         const uint16_t* p_neuron_links,
                                         const uint16_t* p_neuron_internal_links_num,
                                         const uint16_t* p_neuron_external_links_num,
                                         const uint8_t*  p_neuron_act_type_mask,
                                         const flt32_t*  p_weights,
                                         const flt32_t*  p_act_weights,
                                         const flt32_t*  p_neurons_inputs,
                                         const uint16_t  neurons_inputs_num);
 
void nrf_nn_neuton_run_inference_raw_q8(uint8_t*        p_neurons,
                                        const uint32_t  neurons_num,
                                        const uint16_t* p_neuron_links,
                                        const uint16_t* p_neuron_internal_links_num,
                                        const uint16_t* p_neuron_external_links_num,
                                        const uint8_t*  p_neuron_act_type_mask,
                                        const int8_t*   p_weights,
                                        const uint8_t*  p_act_weights,
                                        const uint8_t*  p_neurons_inputs,
                                        const uint16_t  neurons_inputs_num);
 
void nrf_nn_neuton_run_inference_raw_q16(uint16_t*       p_neurons,
                                         const uint32_t  neurons_num,
                                         const uint16_t* p_neuron_links,
                                         const uint16_t* p_neuron_internal_links_num,
                                         const uint16_t* p_neuron_external_links_num,
                                         const uint8_t*  p_neuron_act_type_mask,
                                         const int16_t*  p_weights,
                                         const uint16_t* p_act_weights,
                                         const uint16_t* p_neurons_inputs,
                                         const uint16_t  neurons_inputs_num);
#ifdef __cplusplus
}
#endif
 
#endif /* _NRF_NN_NEUTON_H_ */