24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CPU_DEPTHWISE_CONV2D_NATIVE_KERNEL_H
25 #define ARM_COMPUTE_CPU_DEPTHWISE_CONV2D_NATIVE_KERNEL_H
34 #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
36 #endif // __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
48 using DepthwiseConv2dNativeKernelPtr = std::add_pointer<void(
88 const char *
name()
const override;
102 DepthwiseConv2dNativeKernelPtr _func{
nullptr};
104 bool _has_biases{
false};
SimpleTensor< float > src
decltype(strategy::transforms) typedef type
void configure(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, ITensorInfo *dst, const ConvolutionInfo &info)
Initialize the function's source, destination and parameters.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuDepthwiseConv2dNativeKernel)
std::add_pointer< bool(const DepthwiseConv2dNativeDataTypeISASelectorData &data)>::type DepthwiseConv2dNativeDataTypeISASelectorPtr
static const std::vector< DepthwiseConv2dNativeKernel > & get_available_kernels()
Interface for CPU tensor.
void run_op(ITensorPack &tensors, const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
DepthwiseConv2dNativeKernelPtr ukernel
const Window & window() const
The maximum window the kernel can be executed on.
CpuDepthwiseConv2dNativeKernel()=default
static Status validate(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const ConvolutionInfo &info)
Static function to check if given info will lead to a valid configuration.
Information about executing thread and CPU.
const char * name() const override
Name of the kernel.
const DepthwiseConv2dNativeDataTypeISASelectorPtr is_selected
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
Interface for the kernel to run a depthwise convolution native on a tensor.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)