24 #ifndef __ARM_COMPUTE_CLWINOGRADCONVOLUTIONLAYER_H__ 25 #define __ARM_COMPUTE_CLWINOGRADCONVOLUTIONLAYER_H__ 117 bool _is_activationlayer_enabled;
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.
Basic function to run CLActivationLayerKernel.
Activation Layer Information class.
This file contains all available output stages for GEMMLowp on OpenCL.
Basic function to execute GEMM on OpenCL.
Padding and stride information class.
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.
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...
Basic implementation of the OpenCL tensor interface.