nrf_axon_nn_infer_test.h

Go to the documentation of this file.

/*
 * Copyright (c) 2025 Nordic Semiconductor ASA
 *
 * SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
 */
 
#pragma once
 
#ifdef __cplusplus
extern "C" {
#endif
 
#include <stdarg.h>
#include <stdint.h>
#include "nrf_axon_driver.h"
#include "nrf_axon_nn_infer.h"
 
typedef struct {
        nrf_axon_nn_compiled_model_s base;
        uint8_t layer_ndx;
        int8_t input0_layer_ndx;
        int8_t input1_layer_ndx;
        uint16_t input_batch_cnt;
        uint16_t output_batch_cnt;
 
} nrf_axon_nn_compiled_model_layer_s;
 
#define NRF_AXON_LAYER_MODEL_INPUT_BATCH_CNT(compiled_model_ptr) \
        (compiled_model_ptr->is_layer_model && (compiled_model_ptr->compiler_version >= 0x10209)  ?\
         ((nrf_axon_nn_compiled_model_layer_s *)compiled_model_ptr)->input_batch_cnt : 1)
 
         #define NRF_AXON_LAYER_MODEL_OUTPUT_BATCH_CNT(compiled_model_ptr) \
        (compiled_model_ptr->is_layer_model && (compiled_model_ptr->compiler_version >= 0x10209)  ?\
         ((nrf_axon_nn_compiled_model_layer_s *)compiled_model_ptr)->output_batch_cnt : 1)
 
typedef struct {
        const char *test_name;
        const int8_t **full_model_input_vectors;
        const int8_t **full_model_expected_output_vectors;
        uint16_t full_model_vector_count;
        const int8_t **layer_vectors;
        uint16_t layer_cnt;
} nrf_axon_nn_model_test_info_s;
 
void nrf_axon_nn_populate_model_test_info_s(nrf_axon_nn_model_test_info_s *the_struct,
        const char *test_name,
        const int8_t **full_model_input_vectors,
        const int8_t **full_model_expected_output_vectors,
        uint16_t full_model_vector_count,
        const int8_t **layer_vectors,
        uint16_t layer_cnt);
 
int nrf_axon_nn_run_test_vectors(
        const nrf_axon_nn_compiled_model_s **compiled_full_models,
        const char *test_group_name, uint16_t models_count,
        const nrf_axon_nn_compiled_model_layer_s **compiled_1_layer_models[],
        uint16_t *model_layers_count,
        const nrf_axon_nn_model_test_info_s *test_vectors);
 
int nrf_axon_print_test_inference_results(
        const nrf_axon_nn_compiled_model_s *compiled_model,
        int8_t *output_ptr,
        uint32_t profiling_ticks);
 
void nrf_axon_layer_model_results_to_buffer(
        nrf_axon_nn_compiled_model_layer_s *compiled_layer_model,
        int8_t *to_buffer);
 
#ifdef __cplusplus
} /* extern "C" { */
#endif