diff options
author | Cisco Cervellera <cisco.cervellera@arm.com> | 2021-09-07 11:34:43 +0100 |
---|---|---|
committer | Cisco Cervellera <cisco.cervellera@arm.com> | 2021-09-07 11:34:43 +0100 |
commit | 0210109eb6fa486fda3774e1e74f784f1a32d770 (patch) | |
tree | 6dcbf01ee938d74cfc77e4499e4b9f2176c142d1 /source/application/tensorflow-lite-micro/include/Model.hpp | |
parent | 314948696e0dba0ffc868863868308b027824e1f (diff) | |
download | ml-embedded-evaluation-kit-0210109eb6fa486fda3774e1e74f784f1a32d770.tar.gz |
MLECO-2160: Error repor when Ethos-U is not used
Change-Id: I0dab5308bf5c3eba9b4bb2c9bf0939ac9598d2f6
Diffstat (limited to 'source/application/tensorflow-lite-micro/include/Model.hpp')
-rw-r--r-- | source/application/tensorflow-lite-micro/include/Model.hpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/source/application/tensorflow-lite-micro/include/Model.hpp b/source/application/tensorflow-lite-micro/include/Model.hpp index 70e17ad..b814da4 100644 --- a/source/application/tensorflow-lite-micro/include/Model.hpp +++ b/source/application/tensorflow-lite-micro/include/Model.hpp @@ -84,8 +84,11 @@ namespace app { /** @brief Checks if the model uses signed data. */ bool IsDataSigned() const; - /** @brief Runs the inference (invokes the interpreter). */ - virtual bool RunInference(); + /** @brief Checks if the model uses Ethos-U operator */ + bool ContainsEthosUOperator() const; + + /** @brief Runs the inference (invokes the interpreter). */ + virtual bool RunInference(); /** @brief Model information handler common to all models. * @return true or false based on execution success. |