aboutsummaryrefslogtreecommitdiff
path: root/src/cpu/kernels/CpuWinogradConv2dKernel.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/cpu/kernels/CpuWinogradConv2dKernel.h')
-rw-r--r--src/cpu/kernels/CpuWinogradConv2dKernel.h8
1 files changed, 4 insertions, 4 deletions
diff --git a/src/cpu/kernels/CpuWinogradConv2dKernel.h b/src/cpu/kernels/CpuWinogradConv2dKernel.h
index db2d8acfdb..0c4e28c394 100644
--- a/src/cpu/kernels/CpuWinogradConv2dKernel.h
+++ b/src/cpu/kernels/CpuWinogradConv2dKernel.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2017-2021 Arm Limited.
+ * Copyright (c) 2017-2022 Arm Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -35,7 +35,7 @@ namespace arm_compute
namespace cpu
{
/** Interface for the kernel to perform Winograd input transform. */
-class ICpuWinogradConv2dTransformInputKernel : public ICpuKernel
+class ICpuWinogradConv2dTransformInputKernel : public NewICpuKernel<ICpuWinogradConv2dTransformInputKernel>
{
public:
/** Get the working space required to perform the transformation.
@@ -216,7 +216,7 @@ private:
};
/** Interface for the kernel to perform Winograd output transform. */
-class ICpuWinogradConv2dTransformOutputKernel : public ICpuKernel
+class ICpuWinogradConv2dTransformOutputKernel : public NewICpuKernel<ICpuWinogradConv2dTransformOutputKernel>
{
public:
/** Get the working space required to perform the transformation.
@@ -418,7 +418,7 @@ private:
};
/** Interface for the kernel to perform Winograd weights transform. */
-class ICpuWinogradConv2dTransformWeightsKernel : public ICpuKernel
+class ICpuWinogradConv2dTransformWeightsKernel : public NewICpuKernel<ICpuWinogradConv2dTransformWeightsKernel>
{
public:
/** Prevent instances of this class from being copied (As this class contains pointers) */