24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CPU_DEPTHWISE_CONV2D_ASSEMBLY_DISPATCH_H
25 #define ARM_COMPUTE_CPU_DEPTHWISE_CONV2D_ASSEMBLY_DISPATCH_H
34 struct ConvolutionInfo;
88 std::unique_ptr<LocalImpl> _pImpl;
std::vector< MemoryInfo > MemoryRequirements
SimpleTensor< float > src
Depthwise convolution assembly kernel glue.
void configure(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *bias, ITensorInfo *dst, const ConvolutionInfo &info)
Initialize the function's source, destination, kernels and border_size.
Activation Layer Information class.
CpuDepthwiseConv2dAssemblyDispatch()
void prepare(ITensorPack &tensors) override
Prepare the function for executing.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuDepthwiseConv2dAssemblyDispatch)
Basic interface for functions which have a single async CPU kernel.
static bool is_activation_supported(const ActivationLayerInfo &activation)
Checks if activation is supported by the assembly kernels.
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.
static Status validate(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *bias, const ITensorInfo *dst, const ConvolutionInfo &info)
Static function to check if given info will lead to a valid configuration.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
~CpuDepthwiseConv2dAssemblyDispatch()