diff options
Diffstat (limited to 'source/application/tensorflow-lite-micro/include')
-rw-r--r-- | source/application/tensorflow-lite-micro/include/Model.hpp | 1 | ||||
-rw-r--r-- | source/application/tensorflow-lite-micro/include/TensorFlowLiteMicro.hpp | 13 |
2 files changed, 13 insertions, 1 deletions
diff --git a/source/application/tensorflow-lite-micro/include/Model.hpp b/source/application/tensorflow-lite-micro/include/Model.hpp index b814da4..151b680 100644 --- a/source/application/tensorflow-lite-micro/include/Model.hpp +++ b/source/application/tensorflow-lite-micro/include/Model.hpp @@ -126,7 +126,6 @@ namespace app { size_t GetActivationBufferSize(); private: - tflite::MicroErrorReporter m_uErrorReporter; /* Error reporter object. */ 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. */ diff --git a/source/application/tensorflow-lite-micro/include/TensorFlowLiteMicro.hpp b/source/application/tensorflow-lite-micro/include/TensorFlowLiteMicro.hpp index 545ede1..f6639fd 100644 --- a/source/application/tensorflow-lite-micro/include/TensorFlowLiteMicro.hpp +++ b/source/application/tensorflow-lite-micro/include/TensorFlowLiteMicro.hpp @@ -60,13 +60,26 @@ namespace arm { namespace app { + /** Struct for quantization parameters. */ struct QuantParams { float scale = 1.0; int offset = 0; }; + /** + * @brief Gets the quantization parameters from a tensor + * @param[in] tensor pointer to the tensor. + * @return QuantParams object. + */ QuantParams GetTensorQuantParams(TfLiteTensor* tensor); + /** + * @brief String logging functionality expected to be defined + * by TensorFlow Lite Micro's error reporter. + * @param[in] s Pointer to the string. + */ + extern "C" void DebugLog(const char* s); + } /* namespace app */ } /* namespace arm */ |