21.02
|
Interface for the batch concatenate kernel. More...
#include <CpuConcatenateBatchKernel.h>
Public Member Functions | |
CpuConcatenateBatchKernel () | |
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE (CpuConcatenateBatchKernel) | |
void | configure (const ITensorInfo *src, unsigned int batch_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 batch_offset, const ITensorInfo *dst) |
Static function to check if given info will lead to a valid configuration of CpuConcatenateBatchKernel. More... | |
Interface for the batch concatenate kernel.
The input tensor will be concatenated into the output tensor.
Definition at line 42 of file CpuConcatenateBatchKernel.h.
Definition at line 150 of file CpuConcatenateBatchKernel.cpp.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE | ( | CpuConcatenateBatchKernel | ) |
void configure | ( | const ITensorInfo * | src, |
unsigned int | batch_offset, | ||
ITensorInfo * | dst | ||
) |
Configure kernel for a given list of arguments.
[in] | src | Source tensor info. Data types supported: All. |
[in] | batch_offset | The offset on axis # 3. |
[in,out] | dst | Destination tensor info. Data types supported: Same as src . |
Definition at line 155 of file CpuConcatenateBatchKernel.cpp.
References ARM_COMPUTE_ERROR, ARM_COMPUTE_ERROR_ON_NULLPTR, ARM_COMPUTE_ERROR_THROW_ON, arm_compute::calculate_max_window(), ITensorInfo::data_type(), arm_compute::F16, arm_compute::F32, ITensorInfo::num_dimensions(), arm_compute::QASYMM8, arm_compute::QASYMM8_SIGNED, arm_compute::S16, arm_compute::S32, arm_compute::S8, Dimensions< T >::set_num_dimensions(), ITensorInfo::set_valid_region(), ITensorInfo::tensor_shape(), arm_compute::U16, arm_compute::U32, arm_compute::U8, and arm_compute::validate_arguments().
|
overridevirtual |
Name of the kernel.
Implements ICPPKernel.
Definition at line 214 of file CpuConcatenateBatchKernel.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 201 of file CpuConcatenateBatchKernel.cpp.
References arm_compute::ACL_DST, arm_compute::ACL_SRC, 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().
|
static |
Static function to check if given info will lead to a valid configuration of CpuConcatenateBatchKernel.
[in] | src | Source tensor info. Data types supported: All. |
[in] | batch_offset | The offset on axis # 3. |
[in] | dst | Destination tensor info. Data types supported: Same as src . |
Definition at line 193 of file CpuConcatenateBatchKernel.cpp.
References ARM_COMPUTE_RETURN_ON_ERROR, and arm_compute::validate_arguments().
Referenced by CpuConcatenate::validate().