File ns_model.h

File List > includes-api > ns_model.h

Go to the documentation of this file

#ifndef NS_BASELINE
    #define NS_BASELINE
    #ifdef __cplusplus
        #include "tensorflow/lite/micro/kernels/micro_ops.h"
        #include "tensorflow/lite/micro/micro_interpreter.h"
        #include "tensorflow/lite/micro/micro_mutable_op_resolver.h"
        #include "tensorflow/lite/micro/micro_profiler.h"
        #include "tensorflow/lite/micro/system_setup.h"
        #include "tensorflow/lite/schema/schema_generated.h"
#ifdef NS_TFSTRUCTURE_RECENT
            #include "tensorflow/lite/micro/tflite_bridge/micro_error_reporter.h"
        #else
            #include "tensorflow/lite/micro/micro_error_reporter.h"
        #endif

extern "C" {
    #endif

    #ifdef NS_MLPROFILE
        #include "ns_ambiqsuite_harness.h"
        #include "ns_debug_log.h"
    #endif

typedef enum { READY, NOT_READY, ERROR } ns_model_states_e;
typedef enum { TFLM } ns_model_runtime_e;

    #define NS_MAX_INPUT_TENSORS 3
    #define NS_MAX_OUTPUT_TENSORS 3

typedef struct {
    ns_model_states_e state;

    // Configuration (init by application)
    ns_model_runtime_e runtime; 
    const unsigned char *model_array;
    uint8_t *arena;            
    uint32_t arena_size;       
    uint8_t *rv_arena;         
    uint32_t rv_arena_size;    
    uint32_t rv_count;         
    uint32_t numInputTensors;  
    uint32_t numOutputTensors; 

    #ifdef NS_MLPROFILE
    ns_timer_config_t *tickTimer;       
    ns_perf_mac_count_t *mac_estimates; 
    #else
    void *tickTimer;
    void *mac_estimate;
    #endif
    // State (init by baseline code)
    const tflite::Model *model;                        
    tflite::MicroInterpreter *interpreter;             
    TfLiteTensor *model_input[NS_MAX_INPUT_TENSORS];   
    TfLiteTensor *model_output[NS_MAX_OUTPUT_TENSORS]; 
    tflite::MicroProfiler *profiler;                   
    tflite::ErrorReporter *error_reporter;             

    // metadata
    uint32_t computed_arena_size;
} ns_model_state_t;

extern int ns_model_init(ns_model_state_t *ms);

    #ifdef __cplusplus
}
    #endif
#endif