ARM Compute Library
17.03.1
|
OpenCL kernel which transposes the elements of a matrix in chunks of 1x4 if the input data type is F32 or in chunks of 1x8 if the input data type is F16. More...
#include <CLGEMMTranspose1xWKernel.h>
Public Member Functions | |
void | configure (const ICLTensor *input, ICLTensor *output) |
Initialise the kernel's input and output. More... | |
Public Member Functions inherited from ICLSimple2DKernel | |
void | run (const Window &window, cl::CommandQueue &queue) override |
Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue. More... | |
Public Member Functions inherited from ICLSimpleKernel | |
ICLSimpleKernel () | |
Constructor. More... | |
ICLSimpleKernel (const ICLSimpleKernel &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers). More... | |
ICLSimpleKernel & | operator= (const ICLSimpleKernel &)=delete |
Prevent instances of this class from being copied (As this class contains pointers). More... | |
ICLSimpleKernel (ICLSimpleKernel &&)=default | |
Allow instances of this class to be moved. More... | |
ICLSimpleKernel & | operator= (ICLSimpleKernel &&)=default |
Allow instances of this class to be moved. More... | |
~ICLSimpleKernel ()=default | |
Default destructor. More... | |
void | configure (const ICLTensor *input, ICLTensor *output, unsigned int processed_elements, bool border_undefined=false, const BorderSize &border_size=BorderSize()) |
Configure the kernel. More... | |
Public Member Functions inherited from ICLKernel | |
ICLKernel () | |
Constructor. More... | |
cl::Kernel & | kernel () |
Returns a reference to the OpenCL kernel of this object. More... | |
void | add_1D_tensor_argument (unsigned int &idx, const ICLTensor *tensor, const Window &window) |
Add the passed 1D tensor's parameters to the object's kernel's arguments starting from the index idx. More... | |
void | add_2D_tensor_argument (unsigned int &idx, const ICLTensor *tensor, const Window &window) |
Add the passed 2D tensor's parameters to the object's kernel's arguments starting from the index idx. More... | |
void | add_3D_tensor_argument (unsigned int &idx, const ICLTensor *tensor, const Window &window) |
Add the passed 3D tensor's parameters to the object's kernel's arguments starting from the index idx. More... | |
unsigned int | num_arguments_per_1D_tensor () const |
Returns the number of arguments enqueued per 1D tensor object. More... | |
unsigned int | num_arguments_per_2D_tensor () const |
Returns the number of arguments enqueued per 2D tensor object. More... | |
unsigned int | num_arguments_per_3D_tensor () const |
Returns the number of arguments enqueued per 3D tensor object. More... | |
template<typename T > | |
void | add_argument (unsigned int &idx, T value) |
Add the passed parameters to the object's kernel's arguments starting from the index idx. 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... | |
OpenCL kernel which transposes the elements of a matrix in chunks of 1x4 if the input data type is F32 or in chunks of 1x8 if the input data type is F16.
Following an example of how the transposition1xW works when the input data type is F32
\[ \left( \begin{array}{cccc} a00 & a01 & a02 & a03 \\ a10 & a11 & a12 & a13 \\ a20 & a21 & a22 & a23 \\ a30 & a31 & a32 & a33 \\ \end{array} \right) \rightarrow \left( \begin{array}{ccccccccccccccccc} a00 & a01 & a02 & a03 & a10 & a11 & a12 & a13 & a20 & a21 & a22 & a23 & a30 & a31 & a32 & a33 \\ \end{array} \right) \]
Following an example of how the transposition1xW works when the input data type is F16
\[ \left( \begin{array}{cccccccc} a00 & a01 & a02 & a03 & a04 & a05 & a06 & a7 \\ a10 & a11 & a12 & a13 & a14 & a15 & a16 & 17 \\ a20 & a21 & a22 & a23 & a24 & a25 & a26 & 27 \\ a30 & a31 & a32 & a33 & a34 & a35 & a36 & 37 \\ \end{array} \right) \rightarrow \left( \begin{array}{cccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccc} a00 & a01 & a02 & a03 & a04 & a05 & a06 & a07 & a10 & a11 & a12 & a13 & a14 & a15 & a16 & a17 & a20 & a21 & a22 & a23 & a24 & a25 & a26 & a27 & a30 & a31 & a32 & a33 & a34 & a35 & a36 & a37\\ \end{array} \right) \]
Definition at line 70 of file CLGEMMTranspose1xWKernel.h.
Initialise the kernel's input and output.
[in] | input | Input tensor. Data types supported: U8/F16/F32 |
[out] | output | Output tensor. Data type supported: same as input |