diff options
Diffstat (limited to 'source/application/tensorflow-lite-micro/include/TensorFlowLiteMicro.hpp')
-rw-r--r-- | source/application/tensorflow-lite-micro/include/TensorFlowLiteMicro.hpp | 13 |
1 files changed, 13 insertions, 0 deletions
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 */ |