21.02
|
Interface for the width concatenate kernel. More...
#include <CpuConcatenateWidthKernel.h>
Public Member Functions | |
CpuConcatenateWidthKernel () | |
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE (CpuConcatenateWidthKernel) | |
void | configure (const ITensorInfo *src, unsigned int width_offset, ITensorInfo *dst) |
Configure kernel for a given list of arguments. 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... | |
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... | |
Static Public Member Functions | |
static Status | validate (const ITensorInfo *src, unsigned int width_offset, const ITensorInfo *dst) |
Static function to check if given info will lead to a valid configuration of CpuConcatenateWidthKernel. More... | |
Interface for the width concatenate kernel.
The source tensor will be concatenated into the destination tensor.
Definition at line 40 of file CpuConcatenateWidthKernel.h.
Definition at line 66 of file CpuConcatenateWidthKernel.cpp.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE | ( | CpuConcatenateWidthKernel | ) |
void configure | ( | const ITensorInfo * | src, |
unsigned int | width_offset, | ||
ITensorInfo * | dst | ||
) |
Configure kernel for a given list of arguments.
[in] | src | Source tensor info. Data types supported: All |
[in] | width_offset | The offset on the X axis. |
[in,out] | dst | Destination tensor info. Data types supported: Same as src . |
Definition at line 71 of file CpuConcatenateWidthKernel.cpp.
References ARM_COMPUTE_ERROR_ON_NULLPTR, ARM_COMPUTE_ERROR_THROW_ON, arm_compute::calculate_max_window(), ITensorInfo::num_dimensions(), Dimensions< T >::set_num_dimensions(), ITensorInfo::set_valid_region(), ITensorInfo::tensor_shape(), and arm_compute::validate_arguments().
|
overridevirtual |
Name of the kernel.
Implements ICPPKernel.
Definition at line 177 of file CpuConcatenateWidthKernel.cpp.
|
overridevirtual |
Execute the kernel on the passed window.
[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 93 of file CpuConcatenateWidthKernel.cpp.
References arm_compute::ACL_DST, arm_compute::ACL_SRC, ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW, ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL, ARM_COMPUTE_UNUSED, ITensor::buffer(), arm_compute::dequantize_qasymm8(), arm_compute::dequantize_qasymm8_signed(), Window::DimX, dt, Window::Dimension::end(), arm_compute::execute_window_loop(), ITensorPack::get_const_tensor(), ITensorPack::get_tensor(), Iterator::offset(), Iterator::ptr(), arm_compute::QASYMM8, arm_compute::QASYMM8_SIGNED, arm_compute::quantize_qasymm8(), arm_compute::quantize_qasymm8_signed(), Window::set(), Window::Dimension::start(), arm_compute::vdequantize(), arm_compute::wrapper::vloadq(), arm_compute::vquantize(), arm_compute::vquantize_signed(), arm_compute::wrapper::vstore(), IKernel::window(), and Window::x().
|
static |
Static function to check if given info will lead to a valid configuration of CpuConcatenateWidthKernel.
[in] | src | Source tensor info. Data types supported: All |
[in] | width_offset | The offset on the X axis. |
[in] | dst | Destination tensor info. Data types supported: Same as src . |
Definition at line 87 of file CpuConcatenateWidthKernel.cpp.
References ARM_COMPUTE_RETURN_ON_ERROR, and arm_compute::validate_arguments().
Referenced by CpuConcatenate::validate().