diff options
Diffstat (limited to 'src/backends/reference/workloads/RefElementwiseWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefElementwiseWorkload.hpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/backends/reference/workloads/RefElementwiseWorkload.hpp b/src/backends/reference/workloads/RefElementwiseWorkload.hpp index 065a7833d7..579e5def34 100644 --- a/src/backends/reference/workloads/RefElementwiseWorkload.hpp +++ b/src/backends/reference/workloads/RefElementwiseWorkload.hpp @@ -21,15 +21,15 @@ template <typename Functor, typename ParentDescriptor, typename armnn::StringMap class RefElementwiseWorkload : public RefBaseWorkload<ParentDescriptor> { public: - using InType = typename ElementwiseBinaryFunction<Functor>::InType; - using OutType = typename ElementwiseBinaryFunction<Functor>::OutType; - using RefBaseWorkload<ParentDescriptor>::m_Data; - RefElementwiseWorkload(const ParentDescriptor& descriptor, const WorkloadInfo& info); void Execute() const override; void ExecuteAsync(WorkingMemDescriptor& workingMemDescriptor) override; private: + using InType = typename ElementwiseBinaryFunction<Functor>::InType; + using OutType = typename ElementwiseBinaryFunction<Functor>::OutType; + using RefBaseWorkload<ParentDescriptor>::m_Data; + void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const; }; |