24 #ifndef ARM_COMPUTE_NEWINOGRADCONVOLUTIONLAYER_H 25 #define ARM_COMPUTE_NEWINOGRADCONVOLUTIONLAYER_H 82 bool enable_fast_math =
false);
116 std::unique_ptr<ICPPKernel> _transform_input_kernel;
117 std::unique_ptr<ICPPKernel> _transform_output_kernel;
118 std::unique_ptr<ICPPKernel> _transform_weights_kernel;
124 Tensor _input_transformed;
125 Tensor _output_transformed;
136 bool _is_activationlayer_enabled;
Base class for all functions.
NEWinogradConvolutionLayer & operator=(NEWinogradConvolutionLayer &&)=delete
Prevent instances of this class from being moved (As this class contains non movable objects) ...
Basic function to execute GEMM on Neon.
NEWinogradConvolutionLayer(const std::shared_ptr< IMemoryManager > &memory_manager=nullptr)
Constructor.
Basic function to run CPPPermuteKernel.
Store the tensor's metadata.
void configure(const 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.
Activation Layer Information class.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
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 NEGEMMConvolutionLayer.
~NEWinogradConvolutionLayer()=default
Default destructor.
Basic function to simulate a convolution layer.
Basic implementation of the tensor interface.
Padding and stride information class.
void prepare() override
Prepare the function for executing.
Basic function to run cpu::kernels::CpuActivationKernel.
void run() override
Run the kernels contained in the function.