diff options
author | Matthew Sloyan <matthew.sloyan@arm.com> | 2023-01-12 09:58:07 +0000 |
---|---|---|
committer | Matthew Sloyan <matthew.sloyan@arm.com> | 2023-01-13 12:02:04 +0000 |
commit | 04059e5c4e23085f0a70456c2ff5fe1bc029eb06 (patch) | |
tree | f37d72ad3c068c03346e9be580acb49964cd3143 /src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp | |
parent | 2b32a69f3aac5496d0a966d9740cb4854504f3d9 (diff) | |
download | armnn-04059e5c4e23085f0a70456c2ff5fe1bc029eb06.tar.gz |
IVGCVSW-7381 Add IsLayerSupported implementation to GpuFsa backend
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: Ib18af4a4be4a87e301ff0d0fea5205e985052846
Diffstat (limited to 'src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp')
-rw-r--r-- | src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp b/src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp new file mode 100644 index 0000000000..ecdb3cf597 --- /dev/null +++ b/src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp @@ -0,0 +1,19 @@ +// +// Copyright © 2023 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// +#pragma once + +#include <armnn/Descriptors.hpp> +#include <armnn/Tensor.hpp> + +#include <arm_compute/core/Error.h> + +namespace armnn { + +arm_compute::Status GpuFsaConvolution2dValidate(const TensorInfo& input, + const Convolution2dDescriptor& descriptor, + const TensorInfo& weights, + const Optional<TensorInfo>& biases); + +} // namespace armnn
\ No newline at end of file |