aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/RefNormalizationWorkload.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/RefNormalizationWorkload.hpp')
-rw-r--r--src/backends/reference/workloads/RefNormalizationWorkload.hpp5
1 files changed, 4 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/RefNormalizationWorkload.hpp b/src/backends/reference/workloads/RefNormalizationWorkload.hpp
index 9d68ffda58..59170b8a80 100644
--- a/src/backends/reference/workloads/RefNormalizationWorkload.hpp
+++ b/src/backends/reference/workloads/RefNormalizationWorkload.hpp
@@ -17,7 +17,10 @@ public:
explicit RefNormalizationWorkload(const NormalizationQueueDescriptor& descriptor,
const WorkloadInfo& info);
- virtual void Execute() const override;
+ void Execute() const override;
+ void ExecuteAsync(WorkingMemDescriptor& workingMemDescriptor) override;
+private:
+ void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const;
};
} // namespace armnn