blob: 5026dc3aaf9f587a8488b860e4c4ba3bc9ad7776 [file] [log] [blame]
Mike Kelly831faed2018-11-28 11:52:08 +00001//
Teresa Charlin588cbdf2022-01-19 15:55:37 +00002// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
Mike Kelly831faed2018-11-28 11:52:08 +00003// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
Teresa Charlin588cbdf2022-01-19 15:55:37 +00008#include "ClBaseWorkload.hpp"
Matthew Bentham9b3e7382020-02-05 21:39:55 +00009#include <arm_compute/runtime/CL/functions/CLBatchToSpaceLayer.h>
Mike Kelly831faed2018-11-28 11:52:08 +000010
11namespace armnn
12{
13
14arm_compute::Status ClBatchToSpaceNdWorkloadValidate(const TensorInfo& input,
15 const TensorInfo& output,
Keith Davisbcd860a2021-08-05 14:20:33 +010016 const BatchToSpaceNdDescriptor& descriptor);
Mike Kelly831faed2018-11-28 11:52:08 +000017
Teresa Charlin588cbdf2022-01-19 15:55:37 +000018class ClBatchToSpaceNdWorkload : public ClBaseWorkload<BatchToSpaceNdQueueDescriptor>
Mike Kelly831faed2018-11-28 11:52:08 +000019{
20public:
Sadik Armagane9444752020-12-02 11:28:58 +000021 ClBatchToSpaceNdWorkload(const BatchToSpaceNdQueueDescriptor& descriptor,
22 const WorkloadInfo& info,
23 const arm_compute::CLCompileContext& clCompileContext);
Mike Kelly831faed2018-11-28 11:52:08 +000024
25 void Execute() const override;
26
27private:
28
29 mutable arm_compute::CLBatchToSpaceLayer m_Layer;
30};
31
32} //namespace armnn