diff options
Diffstat (limited to 'src/backends/cl/workloads/ClBatchToSpaceNdWorkload.hpp')
| -rw-r--r-- | src/backends/cl/workloads/ClBatchToSpaceNdWorkload.hpp | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/src/backends/cl/workloads/ClBatchToSpaceNdWorkload.hpp b/src/backends/cl/workloads/ClBatchToSpaceNdWorkload.hpp index 5026dc3aaf..b23c416c4f 100644 --- a/src/backends/cl/workloads/ClBatchToSpaceNdWorkload.hpp +++ b/src/backends/cl/workloads/ClBatchToSpaceNdWorkload.hpp @@ -1,12 +1,15 @@ // -// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017-2018,2020-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once #include "ClBaseWorkload.hpp" +#include "ClWorkloadUtils.hpp" + #include <arm_compute/runtime/CL/functions/CLBatchToSpaceLayer.h> +#include <arm_compute/runtime/CL/functions/CLReshapeLayer.h> namespace armnn { @@ -22,11 +25,14 @@ public: const WorkloadInfo& info, const arm_compute::CLCompileContext& clCompileContext); - void Execute() const override; + virtual void Execute() const override; private: - mutable arm_compute::CLBatchToSpaceLayer m_Layer; + mutable std::unique_ptr<arm_compute::CLReshapeLayer> m_LayerReshapeInput; + mutable std::unique_ptr<arm_compute::CLReshapeLayer> m_LayerReshapeOutput; + arm_compute::CLTensor m_ReshapeInputTensor; + arm_compute::CLTensor m_ReshapeOutputTensor; }; } //namespace armnn |