diff options
author | Kshitij Sisodia <kshitij.sisodia@arm.com> | 2022-05-13 16:44:16 +0100 |
---|---|---|
committer | Kshitij Sisodia <kshitij.sisodia@arm.com> | 2022-05-13 16:44:32 +0100 |
commit | 937052db1df7050f95e14f3ce87a36c3787a7641 (patch) | |
tree | 9baa4a1ef47cba5658adc452fc6f8e7119247ba0 /source/application/api/common/include/Model.hpp | |
parent | 2c2cacd12385ee9cd78504f273b7f7c844118feb (diff) | |
download | ml-embedded-evaluation-kit-937052db1df7050f95e14f3ce87a36c3787a7641.tar.gz |
MLECO-3101: Minor changes for CMSIS pack
Improvements based on feedback from initial tests done with
the generated pack.
Change-Id: Ia769dd6264c53e6b9337473e1c50bdfa6469a216
Signed-off-by: Kshitij Sisodia <kshitij.sisodia@arm.com>
Diffstat (limited to 'source/application/api/common/include/Model.hpp')
-rw-r--r-- | source/application/api/common/include/Model.hpp | 24 |
1 files changed, 12 insertions, 12 deletions
diff --git a/source/application/api/common/include/Model.hpp b/source/application/api/common/include/Model.hpp index df1b259..70c6245 100644 --- a/source/application/api/common/include/Model.hpp +++ b/source/application/api/common/include/Model.hpp @@ -74,7 +74,7 @@ namespace app { **/ bool Init(uint8_t* tensorArenaAddr, uint32_t tensorArenaSize, - uint8_t* nnModelAddr, + const uint8_t* nnModelAddr, uint32_t nnModelSize, tflite::MicroAllocator* allocator = nullptr); @@ -133,17 +133,17 @@ namespace app { size_t GetActivationBufferSize(); private: - tflite::ErrorReporter* m_pErrorReporter = nullptr; /* Pointer to the error reporter. */ - const tflite::Model* m_pModel = nullptr; /* Tflite model pointer. */ - tflite::MicroInterpreter* m_pInterpreter = nullptr; /* Tflite interpreter. */ - tflite::MicroAllocator* m_pAllocator = nullptr; /* Tflite micro allocator. */ - bool m_inited = false; /* Indicates whether this object has been initialised. */ - uint8_t* m_modelAddr = nullptr; /* Model address */ - uint32_t m_modelSize = 0; /* Model size */ - - std::vector<TfLiteTensor*> m_input = {}; /* Model's input tensor pointers. */ - std::vector<TfLiteTensor*> m_output = {}; /* Model's output tensor pointers. */ - TfLiteType m_type = kTfLiteNoType;/* Model's data type. */ + tflite::ErrorReporter* m_pErrorReporter{nullptr}; /* Pointer to the error reporter. */ + const tflite::Model* m_pModel{nullptr}; /* Tflite model pointer. */ + tflite::MicroInterpreter* m_pInterpreter{nullptr}; /* Tflite interpreter. */ + tflite::MicroAllocator* m_pAllocator{nullptr}; /* Tflite micro allocator. */ + bool m_inited{false}; /* Indicates whether this object has been initialised. */ + const uint8_t* m_modelAddr{nullptr}; /* Model address */ + uint32_t m_modelSize{0}; /* Model size */ + + std::vector<TfLiteTensor*> m_input{}; /* Model's input tensor pointers. */ + std::vector<TfLiteTensor*> m_output{}; /* Model's output tensor pointers. */ + TfLiteType m_type{kTfLiteNoType}; /* Model's data type. */ }; } /* namespace app */ |