24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CLWINOGRADCONVOLUTIONLAYER_H
25 #define ARM_COMPUTE_CLWINOGRADCONVOLUTIONLAYER_H
36 class CLCompileContext;
93 bool enable_fast_math =
false);
119 bool enable_fast_math =
false);
145 bool enable_fast_math =
false);
153 std::unique_ptr<Impl> _impl;
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...
Base class for all functions.
Interface for OpenCL tensor.
~CLWinogradConvolutionLayer()
Default Destructor.
Basic function to execute Winograd-based convolution on OpenCL.
void prepare() override
Prepare the function for executing.
Activation Layer Information class.
CLWinogradConvolutionLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default Constructor.
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.
void run() override
Run the kernels contained in the function.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
CLWinogradConvolutionLayer & operator=(const CLWinogradConvolutionLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers)