Interface for the kernel to run a depthwise convolution native on a tensor. More...

#include <CpuDepthwiseConv2dNativeKernel.h>

Collaboration diagram for CpuDepthwiseConv2dNativeKernel:

Data Structures
struct	DepthwiseConv2dNativeKernel

Public Member Functions
	CpuDepthwiseConv2dNativeKernel ()=default

	ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE (CpuDepthwiseConv2dNativeKernel)

void	configure (const ITensorInfo src, const ITensorInfo weights, const ITensorInfo biases, ITensorInfo dst, const ConvolutionInfo &info)
	Initialize the function's source, destination and parameters. More...

void	run_op (ITensorPack &tensors, const Window &window, const ThreadInfo &info) override
	Execute the kernel on the passed window. More...

const char *	name () const override
	Name of the kernel. More...

Public Member Functions inherited from ICPPKernel
virtual	~ICPPKernel ()=default
	Default destructor. More...

virtual void	run (const Window &window, const ThreadInfo &info)
	Execute the kernel on the passed window. More...

virtual void	run_nd (const Window &window, const ThreadInfo &info, const Window &thread_locator)
	legacy compatibility layer for implemantions which do not support thread_locator In these cases we simply narrow the interface down the legacy version More...

virtual size_t	get_mws (const CPUInfo &platform, size_t thread_count) const
	Return minimum workload size of the relevant kernel. More...

Public Member Functions inherited from IKernel
	IKernel ()
	Constructor. More...

virtual	~IKernel ()=default
	Destructor. More...

virtual bool	is_parallelisable () const
	Indicates whether or not the kernel is parallelisable. More...

virtual BorderSize	border_size () const
	The size of the border for that kernel. More...

const Window &	window () const
	The maximum window the kernel can be executed on. More...

bool	is_window_configured () const
	Function to check if the embedded window of this kernel has been configured. More...

Static Public Member Functions
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. More...

static const std::vector< DepthwiseConv2dNativeKernel > &	get_available_kernels ()

Static Public Member Functions inherited from ICpuKernel< CpuDepthwiseConv2dNativeKernel >
static const auto *	get_implementation (const SelectorType &selector, KernelSelectionType selection_type=KernelSelectionType::Supported)
	Micro-kernel selector. More...

Additional Inherited Members
Static Public Attributes inherited from ICPPKernel
static constexpr size_t	default_mws = 1

Detailed Description

Interface for the kernel to run a depthwise convolution native on a tensor.

Definition at line 43 of file CpuDepthwiseConv2dNativeKernel.h.

Constructor & Destructor Documentation

◆ CpuDepthwiseConv2dNativeKernel()

CpuDepthwiseConv2dNativeKernel ( )

default

Member Function Documentation

◆ ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE()

ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE ( CpuDepthwiseConv2dNativeKernel )

◆ configure()

void configure	(	const ITensorInfo *	src,
		const ITensorInfo *	weights,
		const ITensorInfo *	biases,
		ITensorInfo *	dst,
		const ConvolutionInfo &	info
	)

Initialize the function's source, destination and parameters.

Note: Supported data layouts: NHWC

Parameters

[in]	src	Source tensor. DataType supported: QASYMM8/QASYMM8_SIGNED/F16/F32.
[in]	weights	Weights tensor. This is a 3D tensor with dimensions [IFM, W, H]. Data type supported: Same as `src` or QASYMM8/QASYMM8_SIGNED/QSYMM8_PER_CHANNEL when `src` is QASYMM8/QASYMM8_SIGNED.
[in]	biases	Biases tensor. A 1D tensor with dimensions [IFM]. Must be nullptr if not needed. Data type supported: Same as `src`, S32 when src is QASYMM8/QASYMM8_SIGNED.
[out]	dst	Destination tensor. Data type supported: Same as `src`.
[in]	info	Depthwise convolution meta-data.

Definition at line 146 of file CpuDepthwiseConv2dNativeKernel.cpp.

References ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_ERROR_ON_NULLPTR, ARM_COMPUTE_ERROR_THROW_ON, arm_compute::auto_init_if_empty(), arm_compute::calculate_max_window(), ICloneable< T >::clone(), arm_compute::misc::shape_calculator::compute_depthwise_convolution_shape(), ITensorInfo::data_type(), CPUInfo::get(), ICpuKernel< CpuDepthwiseConv2dNativeKernel >::get_implementation(), CPUInfo::get_isa(), arm_compute::test::validation::info, arm_compute::test::validation::output_shape, ITensorInfo::quantization_info(), and arm_compute::cpu::kernels::validate_arguments().

 {
     ARM_COMPUTE_ERROR_ON_NULLPTR(src, weights, dst);
     ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(src, weights, (biases != nullptr) ? biases : nullptr, dst, info));
 
     _has_biases = (biases != nullptr);
     _conv_info  = info;
 
     const auto uk = CpuDepthwiseConv2dNativeKernel::get_implementation(
                         DepthwiseConv2dNativeDataTypeISASelectorData{ weights->data_type(), src->data_type(), CPUInfo::get().get_isa() });
     ARM_COMPUTE_ERROR_ON(uk == nullptr || uk->ukernel == nullptr);
     _func = uk->ukernel;
 
     const TensorShape output_shape = misc::shape_calculator::compute_depthwise_convolution_shape(*src, *weights, info);
     auto_init_if_empty(*dst, src->clone()->set_is_resizable(true).reset_padding().set_tensor_shape(output_shape).set_quantization_info(dst->quantization_info()));
 
     Window win = calculate_max_window(*dst, Steps());
     ICpuKernel::configure(win);
 }

◆ get_available_kernels()

const std::vector< CpuDepthwiseConv2dNativeKernel::DepthwiseConv2dNativeKernel > & get_available_kernels ( )

static

Definition at line 191 of file CpuDepthwiseConv2dNativeKernel.cpp.

 {
     return available_kernels;
 }

◆ name()

const char * name ( ) const

overridevirtual

Name of the kernel.

Returns: Kernel name

Implements ICPPKernel.

Definition at line 186 of file CpuDepthwiseConv2dNativeKernel.cpp.

 {
     return "CpuDepthwiseConv2dNativeKernel";
 }

◆ run_op()

void run_op	(	ITensorPack &	tensors,
		const Window &	window,
		const ThreadInfo &	info
	)

overridevirtual

Execute the kernel on the passed window.

Warning: If is_parallelisable() returns false then the passed window must be equal to window()

Note: The window has to be a region within the window returned by the window() method; The width of the window has to be a multiple of num_elems_processed_per_iteration().

Parameters

[in]	tensors	A vector containing the tensors to operate on.
[in]	window	Region on which to execute the kernel. (Must be a region of the window returned by window())
[in]	info	Info about executing thread and CPU.

Reimplemented from ICPPKernel.

Definition at line 172 of file CpuDepthwiseConv2dNativeKernel.cpp.

References arm_compute::ACL_DST, arm_compute::ACL_SRC_0, arm_compute::ACL_SRC_1, arm_compute::ACL_SRC_2, ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW, ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL, ARM_COMPUTE_UNUSED, ITensorPack::get_const_tensor(), ITensorPack::get_tensor(), and IKernel::window().

 {
     ARM_COMPUTE_UNUSED(info);
     ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL(this);
     ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICpuKernel::window(), window);
     ARM_COMPUTE_ERROR_ON(_func == nullptr);
 
     const auto src     = tensors.get_const_tensor(TensorType::ACL_SRC_0);
     const auto weights = tensors.get_const_tensor(TensorType::ACL_SRC_1);
     const auto biases  = tensors.get_const_tensor(TensorType::ACL_SRC_2);
     auto       dst     = tensors.get_tensor(TensorType::ACL_DST);
     _func(src, weights, biases, dst, window, _has_biases, _conv_info);
 }

◆ validate()

Status validate	(	const ITensorInfo *	src,
		const ITensorInfo *	weights,
		const ITensorInfo *	biases,
		const ITensorInfo *	dst,
		const ConvolutionInfo &	info
	)

static

Static function to check if given info will lead to a valid configuration.

Returns: a status

Definition at line 166 of file CpuDepthwiseConv2dNativeKernel.cpp.

References ARM_COMPUTE_RETURN_ON_ERROR, and arm_compute::cpu::kernels::validate_arguments().

 {
     ARM_COMPUTE_RETURN_ON_ERROR(validate_arguments(src, weights, biases, dst, info));
     return Status{};
 }

The documentation for this class was generated from the following files:

src/cpu/kernels/CpuDepthwiseConv2dNativeKernel.h
src/cpu/kernels/CpuDepthwiseConv2dNativeKernel.cpp

Data Structures

Public Member Functions

Static Public Member Functions

Additional Inherited Members