diff options
Diffstat (limited to 'include/armnn/IStrategy.hpp')
-rw-r--r-- | include/armnn/IStrategy.hpp | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/include/armnn/IStrategy.hpp b/include/armnn/IStrategy.hpp new file mode 100644 index 0000000000..8d29565dcc --- /dev/null +++ b/include/armnn/IStrategy.hpp @@ -0,0 +1,31 @@ +// +// Copyright © 2021 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// +#pragma once + +#include <armnn/DescriptorsFwd.hpp> +#include <armnn/Types.hpp> + +namespace armnn +{ + +class IStrategy +{ +protected: +IStrategy() {} +virtual ~IStrategy() {} + +public: +virtual void ExecuteStrategy(const armnn::IConnectableLayer* layer, + const armnn::BaseDescriptor& descriptor, + const std::vector<armnn::ConstTensor>& constants, + const char* name, + const armnn::LayerBindingId id = 0) = 0; + +virtual void FinishStrategy() {}; + +}; + + +} // namespace armnn |