21.02
|
Interface for the height concatenate kernel. More...
#include <CpuConcatenateHeightKernel.h>
Public Member Functions | |
CpuConcatenateHeightKernel () | |
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE (CpuConcatenateHeightKernel) | |
void | configure (const ITensorInfo *src, unsigned int height_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... | |
![]() | |
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... | |
![]() | |
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 height_offset, const ITensorInfo *dst) |
Static function to check if given info will lead to a valid configuration of CpuConcatenateHeightKernel. More... | |
Interface for the height concatenate kernel.
The source tensor will be concatenated into the destination tensor.
Definition at line 39 of file CpuConcatenateHeightKernel.h.
Definition at line 66 of file CpuConcatenateHeightKernel.cpp.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE | ( | CpuConcatenateHeightKernel | ) |
void configure | ( | const ITensorInfo * | src, |
unsigned int | height_offset, | ||
ITensorInfo * | dst | ||
) |
Configure kernel for a given list of arguments.
[in] | src | Source tensor info. Data types supported: All |
[in] | height_offset | The starting offset on the Y axis for the output tensor. |
[in,out] | dst | Destination tensor info. Data types supported: Same as src . |
Definition at line 71 of file CpuConcatenateHeightKernel.cpp.
References ARM_COMPUTE_ERROR_ON_NULLPTR, ARM_COMPUTE_ERROR_THROW_ON, ARM_COMPUTE_UNUSED, 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 181 of file CpuConcatenateHeightKernel.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 CpuConcatenateHeightKernel.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, Window::DimY, 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 CpuConcatenateHeightKernel.
[in] | src | Source tensor info. Data types supported: All |
[in] | height_offset | The starting offset on the Y axis for the output tensor. |
[in] | dst | Destination tensor info. Data types supported: Same as src . |
Definition at line 87 of file CpuConcatenateHeightKernel.cpp.
References ARM_COMPUTE_RETURN_ON_ERROR, and arm_compute::validate_arguments().
Referenced by CpuConcatenate::validate().