24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_ARM_COMPUTE_RUNTIME_NEON_FUNCTIONS_NEWINOGRADCONVOLUTIONLAYER_H
25 #define ACL_ARM_COMPUTE_RUNTIME_NEON_FUNCTIONS_NEWINOGRADCONVOLUTIONLAYER_H
96 bool enable_fast_math =
false);
114 bool enable_fast_math =
false);
118 std::unique_ptr<Impl> _impl;
121 #endif // ACL_ARM_COMPUTE_RUNTIME_NEON_FUNCTIONS_NEWINOGRADCONVOLUTIONLAYER_H
Basic function to simulate a convolution layer.
Base class for all functions.
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 NEWinogradConvolutionLay...
Interface for CPU tensor.
NEWinogradConvolutionLayer(const std::shared_ptr< IMemoryManager > &memory_manager=nullptr)
Constructor.
Activation Layer Information class.
NEWinogradConvolutionLayer & operator=(const NEWinogradConvolutionLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
void prepare() override
Prepare the function for executing.
Copyright (c) 2017-2024 Arm Limited.
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.
Store the tensor's metadata.
void run() override
Run the kernels contained in the function.
~NEWinogradConvolutionLayer()
Destructor.