diff options
Diffstat (limited to 'src/dynamic/sample/SampleDynamicWorkloadFactory.hpp')
| -rw-r--r-- | src/dynamic/sample/SampleDynamicWorkloadFactory.hpp | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/src/dynamic/sample/SampleDynamicWorkloadFactory.hpp b/src/dynamic/sample/SampleDynamicWorkloadFactory.hpp index a5a31e5d6e..8cd36c5518 100644 --- a/src/dynamic/sample/SampleDynamicWorkloadFactory.hpp +++ b/src/dynamic/sample/SampleDynamicWorkloadFactory.hpp @@ -52,14 +52,18 @@ public: std::unique_ptr<armnn::IWorkload> CreateAddition( const armnn::AdditionQueueDescriptor& descriptor, - const armnn::WorkloadInfo& info) const override; + const armnn::WorkloadInfo& info) const; std::unique_ptr<armnn::IWorkload> CreateInput(const armnn::InputQueueDescriptor& descriptor, - const armnn::WorkloadInfo& info) const override; + const armnn::WorkloadInfo& info) const; std::unique_ptr<armnn::IWorkload> CreateOutput(const armnn::OutputQueueDescriptor& descriptor, - const armnn::WorkloadInfo& info) const override; + const armnn::WorkloadInfo& info) const; + + std::unique_ptr<armnn::IWorkload> CreateWorkload(armnn::LayerType type, + const armnn::QueueDescriptor& descriptor, + const armnn::WorkloadInfo& info) const override; private: mutable std::shared_ptr<SampleMemoryManager> m_MemoryManager; |