diff options
Diffstat (limited to 'src/backends/reference/workloads/RefSpaceToBatchNdWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefSpaceToBatchNdWorkload.hpp | 6 |
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; }; |