diff options
Diffstat (limited to 'include/armnn/IStrategy.hpp')
-rw-r--r-- | include/armnn/IStrategy.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/include/armnn/IStrategy.hpp b/include/armnn/IStrategy.hpp index 8d29565dcc..aed881c0dd 100644 --- a/include/armnn/IStrategy.hpp +++ b/include/armnn/IStrategy.hpp @@ -7,6 +7,9 @@ #include <armnn/DescriptorsFwd.hpp> #include <armnn/Types.hpp> +#include <armnn/NetworkFwd.hpp> +#include <armnn/TensorFwd.hpp> + namespace armnn { @@ -17,7 +20,7 @@ IStrategy() {} virtual ~IStrategy() {} public: -virtual void ExecuteStrategy(const armnn::IConnectableLayer* layer, +virtual void ExecuteStrategy(const IConnectableLayer* layer, const armnn::BaseDescriptor& descriptor, const std::vector<armnn::ConstTensor>& constants, const char* name, |