24 #ifndef ARM_COMPUTE_NEFFTCONVOLUTIONLAYER_H 25 #define ARM_COMPUTE_NEFFTCONVOLUTIONLAYER_H 125 std::unique_ptr<NEFFT2D> _transform_weights_func;
126 NEFFT2D _itransform_output_func;
141 Tensor _transformed_input;
142 Tensor _transformed_weights;
143 Tensor _input_weights_product;
146 Tensor _itransformed_output;
150 const ITensor *_original_weights;
152 bool _is_activationlayer_enabled;
void prepare() override
Prepare the function for executing.
Base class for all functions.
Basic function to run cpu::kernels::CpuAddKernel.
Basic function to run NEComplexPixelWiseMultiplicationKernel.
Basic function to perform tensor slicing.
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 NEFFTConvolutionLayer.
Basic function to run cpu::kernels::CpuPermuteKernel.
Activation Layer Information class.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
NEFFTConvolutionLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default constructor.
Basic function to execute two dimensional FFT.
Basic implementation of the tensor interface.
Padding and stride information class.
Basic function to run cpu::kernels::CpuActivationKernel.
void run() override
Run the kernels contained in the function.
NEFFTConvolutionLayer & operator=(const NEFFTConvolutionLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Basic function to pad a tensor.
void configure(ITensor *input, const ITensor *weights, const ITensor *biases, ITensor *output, const PadStrideInfo &conv_info, const ActivationLayerInfo &act_info=ActivationLayerInfo(), bool enable_fast_math=false)
Set the input and output tensors.
Basic function to simulate a reduction operation.
Basic function to run NEReverseKernel.
~NEFFTConvolutionLayer()
Default destructor.
Basic function to execute FFT-based convolution on Neon.