24 #ifndef ARM_COMPUTE_CLFFTCONVOLUTIONLAYER_H 25 #define ARM_COMPUTE_CLFFTCONVOLUTIONLAYER_H 153 std::unique_ptr<CLFFT2D> _transform_weights_func;
154 CLFFT2D _itransform_output_func;
180 bool _is_activationlayer_enabled;
Basic function to execute FFT-based convolution on OpenCL.
Basic function to run CLReverseKernel.
Base class for all functions.
void run() override
Run the kernels contained in the function.
Basic function to pad a tensor.
Basic function to run opencl::ClComplexMul.
Basic function to run opencl::kernels::ClSaturatedArithmeticKernel for addition.
Store the tensor's metadata.
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 CLFFTConvolutionLayer.
Basic function to run opencl::kernels::ClActivationKernel.
Activation Layer Information class.
Basic function to perform tensor slicing.
Copyright (c) 2017-2021 Arm Limited.
Basic function to execute an opencl::kernels::ClPermuteKernel.
Padding and stride information class.
Basic function to execute two dimensional FFT.
Interface for OpenCL tensor.
CLFFTConvolutionLayer & operator=(const CLFFTConvolutionLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Perform reduction operation.
CLFFTConvolutionLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default constructor.
void prepare() override
Prepare the function for executing.
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.
Basic implementation of the OpenCL tensor interface.