24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CPUWINOGRADCONV2DKERNEL_H
25 #define ARM_COMPUTE_CPUWINOGRADCONV2DKERNEL_H
34 #include "src/core/NEON/kernels/assembly/winograd.hpp"
35 #include "src/core/NEON/kernels/convolution/common/tensor.hpp"
58 arm_conv::ConvolutionArgs &_c_args,
64 const char *
name()
const override
66 return "CpuWinogradConv2dTransformInputKernel";
70 arm_conv::winograd::WinogradImpl &_winograd_impl;
71 arm_conv::ConvolutionArgs &_conv_args;
90 arm_conv::ConvolutionArgs &_c_args,
96 const char *
name()
const override
98 return "CpuWinogradConv2dTransformOutputKernel";
102 arm_conv::winograd::WinogradImpl &_winograd_impl;
103 const arm_conv::ConvolutionArgs &_conv_args;
const Window & window() const
The maximum window the kernel can be executed on.
Information about executing thread and CPU.
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)