diff options
author | Declan-ARM <decmce01@arm.com> | 2024-02-07 13:07:31 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2024-02-09 14:01:11 +0000 |
commit | 5e90b831fc31605b5734531387837123fb31d0e0 (patch) | |
tree | c75cedfaa5f6170262eb3dfcea6e0522d89e3433 /src/backends/gpuFsa/layers/GpuFsaReshape.hpp | |
parent | ddbda6a1ed13f7bef7e0dce07a37e91b062ce98a (diff) | |
download | armnn-5e90b831fc31605b5734531387837123fb31d0e0.tar.gz |
IVGCVSW-7569 GpuFsa Op: Add Reshape Operator
* Add Reshape EndToEnd tests to all backends
Signed-off-by: Declan-ARM <decmce01@arm.com>
Change-Id: Ic6d07ba8de0cf3271ed0e4c6d604e070ccb968e3
Diffstat (limited to 'src/backends/gpuFsa/layers/GpuFsaReshape.hpp')
-rw-r--r-- | src/backends/gpuFsa/layers/GpuFsaReshape.hpp | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/src/backends/gpuFsa/layers/GpuFsaReshape.hpp b/src/backends/gpuFsa/layers/GpuFsaReshape.hpp new file mode 100644 index 0000000000..16fa1f975e --- /dev/null +++ b/src/backends/gpuFsa/layers/GpuFsaReshape.hpp @@ -0,0 +1,21 @@ +// +// Copyright © 2024 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include <armnn/Descriptors.hpp> +#include <gpuFsa/GpuFsaBackend.hpp> + +namespace armnn +{ + +arm_compute::Status GpuFsaReshapeValidate(const TensorInfo& input, const ReshapeDescriptor& descriptor); + +void GpuFsaReshapeCreateOp(GpuFsaPreCompiledBlob* blob, + const TensorInfo& input, + const ReshapeDescriptor& descriptor); + +} // namespace armnn + |