ARM Compute Library
17.04
|
NEON kernel to multiply matrices. More...
#include <NEGEMMLowpMatrixMultiplyKernel.h>
Public Member Functions | |
NEGEMMLowpMatrixMultiplyKernel () | |
Constructor. More... | |
NEGEMMLowpMatrixMultiplyKernel (const NEGEMMLowpMatrixMultiplyKernel &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEGEMMLowpMatrixMultiplyKernel & | operator= (const NEGEMMLowpMatrixMultiplyKernel &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEGEMMLowpMatrixMultiplyKernel (NEGEMMLowpMatrixMultiplyKernel &&)=default | |
Allow instances of this class to be moved. More... | |
NEGEMMLowpMatrixMultiplyKernel & | operator= (NEGEMMLowpMatrixMultiplyKernel &&)=default |
Allow instances of this class to be moved. More... | |
void | configure (const ITensor *input0, const ITensor *input1, ITensor *output, int32_t a_offset, int32_t b_offset, int32_t output_offset, int32_t output_mult_int, int32_t shift) |
Initialise the kernel's input and output. More... | |
void | run (const Window &window) override |
Execute the kernel on the passed window. More... | |
Public Member Functions inherited from ICPPKernel | |
virtual | ~ICPPKernel ()=default |
Default destructor. 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... | |
NEON kernel to multiply matrices.
Definition at line 46 of file NEGEMMLowpMatrixMultiplyKernel.h.
Constructor.
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
default |
Allow instances of this class to be moved.
void configure | ( | const ITensor * | input0, |
const ITensor * | input1, | ||
ITensor * | output, | ||
int32_t | a_offset, | ||
int32_t | b_offset, | ||
int32_t | output_offset, | ||
int32_t | output_mult_int, | ||
int32_t | shift | ||
) |
Initialise the kernel's input and output.
The input matrices input0
and input1
must be the output of the kernels: NEGEMMInterleave4x4Kernel and NEGEMMTranspose1xWKernel. These two kernels change the layout of the original matrices to be more cache-friendly.
[in] | input0 | Input tensor containing the interleaved Matrix A. Data type supported: U8 |
[in] | input1 | Input tensor containing the transposed Matrix B. Data type supported: same as input0 |
[out] | output | Output tensor to store the result of matrix multiplication, Data type supported: same as input0 |
[in] | a_offset | Offset to be added to each element of the matrix A. |
[in] | b_offset | Offset to be added to each element of the matrix B. |
[in] | output_offset | Offset to be added to each element of the output matrix |
[in] | output_mult_int | Value to be multipied to each entry of the result. |
[in] | shift | Number of bits to shift right the result. |
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
default |
Allow instances of this class to be moved.
|
overridevirtual |
Execute the kernel on the passed window.
[in] | window | Region on which to execute the kernel. (Must be a region of the window returned by window()) |
Implements ICPPKernel.