diff options
author | Colm Donelan <Colm.Donelan@arm.com> | 2021-10-12 15:06:19 +0100 |
---|---|---|
committer | Matthew Sloyan <matthew.sloyan@arm.com> | 2021-10-18 10:42:15 +0000 |
commit | 3cff15a5d8797d0afe5d5b1cb3ff0e5b7d1cd6c9 (patch) | |
tree | 47a203daa2fee1f7afc3e79787cf2e99c52473da /tests/ExecuteNetwork/ExecuteNetworkParams.hpp | |
parent | b4128b4c8c8c717225c83debc5f957d37d6513fa (diff) | |
download | armnn-3cff15a5d8797d0afe5d5b1cb3ff0e5b7d1cd6c9.tar.gz |
IVGCVSW-5879 Pass the execute network parameters to the TfLiteDelegate.
* Introduce a mechanism to construct a DelegateOptions from an
ExecuteNetworkParams.
* Modify ExecuteNetwork to use this constructed DelegateOptions.
Signed-off-by: Colm Donelan <Colm.Donelan@arm.com>
Change-Id: Ied663a1e00ac3eece42244ed313ddafd6d2ce078
Diffstat (limited to 'tests/ExecuteNetwork/ExecuteNetworkParams.hpp')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetworkParams.hpp | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetworkParams.hpp b/tests/ExecuteNetwork/ExecuteNetworkParams.hpp index db8194b3f9..cb8c2fb386 100644 --- a/tests/ExecuteNetwork/ExecuteNetworkParams.hpp +++ b/tests/ExecuteNetwork/ExecuteNetworkParams.hpp @@ -8,6 +8,10 @@ #include <armnn/BackendId.hpp> #include <armnn/Tensor.hpp> +#if defined(ARMNN_TFLITE_DELEGATE) +#include <DelegateOptions.hpp> +#endif + /// Holds all parameters necessary to execute a network /// Check ExecuteNetworkProgramOptions.cpp for a description of each parameter struct ExecuteNetworkParams @@ -64,4 +68,10 @@ struct ExecuteNetworkParams // Ensures that the parameters for ExecuteNetwork fit together void ValidateParams(); + +#if defined(ARMNN_TFLITE_DELEGATE) + /// A utility method that populates a DelegateOptions object from this ExecuteNetworkParams. + armnnDelegate::DelegateOptions ToDelegateOptions() const; +#endif + }; |