aboutsummaryrefslogtreecommitdiff
path: root/src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp
diff options
context:
space:
mode:
authorCathal Corbett <cathal.corbett@arm.com>2023-01-13 19:19:02 +0000
committerCathal Corbett <cathal.corbett@arm.com>2023-01-17 16:53:29 +0000
commitd6161b4f1bad0fc0095e73747159480066f56edb (patch)
treebeee339a1532fc81391409ff9f74c5a3723a1749 /src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp
parent04059e5c4e23085f0a70456c2ff5fe1bc029eb06 (diff)
downloadarmnn-experimental/GpuFsa.tar.gz
IVGCVSW-7382 Implementation of Conv2d within GpuFsaexperimental/GpuFsa
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com> Change-Id: I6802687bb959c74afb1b9aebed133b2b17c036dc
Diffstat (limited to 'src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp')
-rw-r--r--src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp12
1 files changed, 11 insertions, 1 deletions
diff --git a/src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp b/src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp
index ecdb3cf597..79a2fec400 100644
--- a/src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp
+++ b/src/backends/gpuFsa/layerValidators/GpuFsaConvolution2dValidate.hpp
@@ -8,12 +8,22 @@
#include <armnn/Tensor.hpp>
#include <arm_compute/core/Error.h>
+#include <arm_compute/dynamic_fusion/sketch/gpu/GpuWorkloadSketch.h>
-namespace armnn {
+namespace armnn
+{
+
+using namespace arm_compute::experimental::dynamic_fusion;
arm_compute::Status GpuFsaConvolution2dValidate(const TensorInfo& input,
const Convolution2dDescriptor& descriptor,
const TensorInfo& weights,
const Optional<TensorInfo>& biases);
+void GpuFsaConvolution2dCreateOp(const TensorInfo& input,
+ const Convolution2dDescriptor& descriptor,
+ const TensorInfo& weights,
+ const Optional<TensorInfo>& biases,
+ GpuWorkloadSketch& sketch);
+
} // namespace armnn \ No newline at end of file