diff options
Diffstat (limited to 'src/armnn/backends/NeonWorkloads/NeonBaseSplitterWorkload.hpp')
-rw-r--r-- | src/armnn/backends/NeonWorkloads/NeonBaseSplitterWorkload.hpp | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/src/armnn/backends/NeonWorkloads/NeonBaseSplitterWorkload.hpp b/src/armnn/backends/NeonWorkloads/NeonBaseSplitterWorkload.hpp index 769905b48b..769291c700 100644 --- a/src/armnn/backends/NeonWorkloads/NeonBaseSplitterWorkload.hpp +++ b/src/armnn/backends/NeonWorkloads/NeonBaseSplitterWorkload.hpp @@ -6,20 +6,21 @@ #pragma once #include <backends/Workload.hpp> +#include <backends/NeonWorkloadUtils.hpp> namespace armnn { -// Base class template providing an implementation of the Splitter layer common to all data types -template <armnn::DataType DataType> -class NeonBaseSplitterWorkload : public TypedWorkload<SplitterQueueDescriptor, DataType> +// Base class template providing an implementation of the Splitter layer common to all data types. +template <armnn::DataType... DataTypes> +class NeonBaseSplitterWorkload : public TypedWorkload<SplitterQueueDescriptor, DataTypes...> { public: - using TypedWorkload<SplitterQueueDescriptor, DataType>::TypedWorkload; + using TypedWorkload<SplitterQueueDescriptor, DataTypes...>::TypedWorkload; virtual void Execute() const override { - // With subtensors, splitter is a no-op + // With subtensors, splitter is a no-op. } }; |