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_ */