Edge AI Add-on API 2.2.0
Loading...
Searching...
No Matches
nrf_axon_nn_infer_test.h File Reference
#include <stdarg.h>
#include <stdint.h>
#include "nrf_axon_driver.h"
#include "nrf_axon_nn_infer.h"

Go to the source code of this file.

Data Structures

struct  nrf_axon_nn_compiled_model_layer_s
 
struct  nrf_axon_nn_model_test_info_s
 

Macros

#define NRF_AXON_LAYER_MODEL_INPUT_BATCH_CNT(compiled_model_ptr)
 
#define NRF_AXON_LAYER_MODEL_OUTPUT_BATCH_CNT(compiled_model_ptr)
 

Functions

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)
 Populates a nrf_axon_nn_model_test_info_s instance with the provided data.
 
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)
 Run the inference test vectors for the passed models.
 
int nrf_axon_print_test_inference_results (const nrf_axon_nn_compiled_model_s *compiled_model, int8_t *output_ptr, uint32_t profiling_ticks)
 Prints formatted test result.
 
void nrf_axon_layer_model_results_to_buffer (nrf_axon_nn_compiled_model_layer_s *compiled_layer_model, int8_t *to_buffer)
 Copies a padded layer model output to an unpadded buffer.