ArmNN
 25.11
Loading...
Searching...
No Matches
NeonDepthToSpaceWorkload.hpp
Go to the documentation of this file.
1//
2// Copyright © 2019 Arm Ltd and Contributors. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
9
10#include "NeonBaseWorkload.hpp"
11
12#include <arm_compute/runtime/NEON/functions/NEDepthToSpaceLayer.h>
13
14namespace armnn
15{
16
17arm_compute::Status NeonDepthToSpaceWorkloadValidate(const TensorInfo& input,
18 const TensorInfo& output,
19 const DepthToSpaceDescriptor& descriptor);
20
21class NeonDepthToSpaceWorkload : public NeonBaseWorkload<DepthToSpaceQueueDescriptor>
22{
23public:
25 void Execute() const override;
26
27private:
28 mutable arm_compute::NEDepthToSpaceLayer m_Layer;
29};
30
31} // namespace armnn
NeonBaseWorkload(const DepthToSpaceQueueDescriptor &descriptor, const WorkloadInfo &info)
NeonDepthToSpaceWorkload(const DepthToSpaceQueueDescriptor &descriptor, const WorkloadInfo &info)
Copyright (c) 2021 ARM Limited and Contributors.
arm_compute::Status NeonDepthToSpaceWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const DepthToSpaceDescriptor &descriptor)
SpaceToDepthDescriptor DepthToSpaceDescriptor
A DepthToSpaceDescriptor for the DepthToSpaceLayer.
Contains information about TensorInfos of a layer.