aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/RefSpaceToBatchNdWorkload.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/RefSpaceToBatchNdWorkload.hpp')
-rw-r--r--src/backends/reference/workloads/RefSpaceToBatchNdWorkload.hpp6
1 files changed, 3 insertions, 3 deletions
diff --git a/src/backends/reference/workloads/RefSpaceToBatchNdWorkload.hpp b/src/backends/reference/workloads/RefSpaceToBatchNdWorkload.hpp
index f2c87682db..f9d75ee4d6 100644
--- a/src/backends/reference/workloads/RefSpaceToBatchNdWorkload.hpp
+++ b/src/backends/reference/workloads/RefSpaceToBatchNdWorkload.hpp
@@ -1,13 +1,11 @@
//
-// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
+// Copyright © 2018-2023 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
#include "RefBaseWorkload.hpp"
-#include <armnn/TypesUtils.hpp>
-
namespace armnn
{
@@ -15,8 +13,10 @@ class RefSpaceToBatchNdWorkload : public RefBaseWorkload<SpaceToBatchNdQueueDesc
{
public:
using RefBaseWorkload<SpaceToBatchNdQueueDescriptor>::RefBaseWorkload;
+
void Execute() const override;
void ExecuteAsync(ExecutionData& executionData) override;
+
private:
void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const;
};