diff options
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 + }; |