blob: 86edd90dc149d1458a04a2c491609cba86e006e9 [file] [log] [blame]
Sadik Armagan48f011e2021-04-21 10:50:34 +01001//
2// Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
Colm Donelan0c479742021-12-10 12:43:54 +00008#include <armnn/backends/Workload.hpp>
Sadik Armagan48f011e2021-04-21 10:50:34 +01009
10#include <arm_compute/core/Error.h>
11#include <arm_compute/runtime/NEON/functions/NECast.h>
12
13namespace armnn
14{
15
16arm_compute::Status NeonCastValidate(const TensorInfo& input, const TensorInfo& output);
17
18class NeonCastWorkload : public BaseWorkload<CastQueueDescriptor>
19{
20public:
21 NeonCastWorkload(const CastQueueDescriptor& descriptor, const WorkloadInfo& info);
22 virtual void Execute() const override;
23
24private:
25 mutable arm_compute::NECast m_CastLayer;
26};
27
28} // namespace armnn