aboutsummaryrefslogtreecommitdiff
path: root/src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp
diff options
context:
space:
mode:
authorMatthew Sloyan <matthew.sloyan@arm.com>2023-01-12 09:58:07 +0000
committerMatthew Sloyan <matthew.sloyan@arm.com>2023-01-13 12:02:04 +0000
commit04059e5c4e23085f0a70456c2ff5fe1bc029eb06 (patch)
treef37d72ad3c068c03346e9be580acb49964cd3143 /src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp
parent2b32a69f3aac5496d0a966d9740cb4854504f3d9 (diff)
downloadarmnn-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.hpp19
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