24 #ifndef ARM_COMPUTE_CLWINOGRADCONVOLUTIONLAYER_H 25 #define ARM_COMPUTE_CLWINOGRADCONVOLUTIONLAYER_H 34 class CLCompileContext;
35 class CLWinogradFilterTransformKernel;
36 class CLWinogradOutputTransformKernel;
132 std::unique_ptr<CLWinogradFilterTransformKernel> _filter_transform;
133 std::unique_ptr<CLWinogradOutputTransformKernel> _output_transform;
Base class for all functions.
CLWinogradConvolutionLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default constructor.
void run() override
Run the kernels contained in the function.
void prepare() override
Prepare the function for executing.
Store the tensor's metadata.
Activation Layer Information class.
Copyright (c) 2017-2021 Arm Limited.
Basic function to execute GEMM on OpenCL.
Padding and stride information class.
static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const PadStrideInfo &conv_info, const ActivationLayerInfo &act_info=ActivationLayerInfo(), bool enable_fast_math=false)
Static function to check if given info will lead to a valid configuration of CLWinogradConvolutionLay...
~CLWinogradConvolutionLayer()
Default destructor.
Interface for OpenCL tensor.
void configure(ICLTensor *input, const ICLTensor *weights, const ICLTensor *biases, ICLTensor *output, const PadStrideInfo &conv_info, const ActivationLayerInfo &act_info=ActivationLayerInfo(), bool enable_fast_math=false)
Set the input and output tensors.
CLWinogradConvolutionLayer & operator=(const CLWinogradConvolutionLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Basic function to execute Winograd-based convolution on OpenCL.
Basic implementation of the OpenCL tensor interface.