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