diff options
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. |