24 #ifndef ARM_COMPUTE_NEDEPTHWISECONVOLUTIONASSEMBLYDISPATCH_H 25 #define ARM_COMPUTE_NEDEPTHWISECONVOLUTIONASSEMBLYDISPATCH_H 120 std::unique_ptr<LocalImpl> _pImpl;
Base class for all functions.
static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *bias, const ITensorInfo *output, const PadStrideInfo &conv_info, unsigned int depth_multiplier=1, const ActivationLayerInfo &act_info=ActivationLayerInfo(), const Size2D &dilation=Size2D(1, 1))
Static function to check if given info will lead to a valid configuration of NEDepthwiseConvolutionAs...
void prepare() override
Prepare the function for executing.
Store the tensor's metadata.
Activation Layer Information class.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
NEDepthwiseConvolutionAssemblyDispatch(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default constructor.
void configure(const ITensor *input, const ITensor *weights, const ITensor *bias, ITensor *output, const PadStrideInfo &conv_info, unsigned int depth_multiplier=1, const ActivationLayerInfo &act_info=ActivationLayerInfo(), const Size2D &dilation=Size2D(1, 1))
Initialize the function's source, destination, kernels and border_size.
Basic implementation of the tensor interface.
Padding and stride information class.
static bool is_optimized_supported(const ITensorInfo *input, const ITensorInfo *weights, PadStrideInfo conv_info, unsigned int depth_multiplier=1, const Size2D &dilation=Size2D(1, 1))
Check if the optimized kernel can be used for the given kernel sizes and strides. ...
Class for specifying the size of an image or rectangle.
Depthwise convolution assembly kernel glue.
NEDepthwiseConvolutionAssemblyDispatch & operator=(const NEDepthwiseConvolutionAssemblyDispatch &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
void run() override
Run the kernels contained in the function.
~NEDepthwiseConvolutionAssemblyDispatch()
Default destructor.