24 #ifndef ARM_COMPUTE_CLWINOGRADCONVOLUTIONLAYER_H 25 #define ARM_COMPUTE_CLWINOGRADCONVOLUTIONLAYER_H 35 class CLCompileContext;
136 std::unique_ptr<Impl> _impl;
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-2022 Arm Limited.
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.