Compute Library
18.05
|
Base class for GEMM NEON kernels implemented in Assembly. More...
#include <NEGEMMAssemblyBaseKernel.h>
Public Member Functions | |
const char * | name () const override |
Name of the kernel. More... | |
NEGEMMAssemblyBaseKernel () | |
Constructor. More... | |
NEGEMMAssemblyBaseKernel (const NEGEMMAssemblyBaseKernel &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEGEMMAssemblyBaseKernel & | operator= (const NEGEMMAssemblyBaseKernel &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEGEMMAssemblyBaseKernel (NEGEMMAssemblyBaseKernel &&)=default | |
Allow instances of this class to be moved. More... | |
NEGEMMAssemblyBaseKernel & | operator= (NEGEMMAssemblyBaseKernel &&)=default |
Allow instances of this class to be moved. More... | |
virtual | ~NEGEMMAssemblyBaseKernel ()=default |
void | configure (const ITensor *input0, const ITensor *input1, ITensor *output, ITensor *workspace, float alpha=1.f, float beta=0.f, bool is_transposed_0=false, bool is_transposed_1=false) |
Initialise the kernel's input and output. More... | |
Public Member Functions inherited from ICPPKernel | |
virtual | ~ICPPKernel ()=default |
Default destructor. More... | |
virtual void | run (const Window &window, const ThreadInfo &info)=0 |
Execute the kernel on the passed window. 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... | |
Base class for GEMM NEON kernels implemented in Assembly.
Definition at line 34 of file NEGEMMAssemblyBaseKernel.h.
|
inline |
Constructor.
Definition at line 42 of file NEGEMMAssemblyBaseKernel.h.
References NEGEMMAssemblyBaseKernel::operator=(), and NEGEMMAssemblyBaseKernel::~NEGEMMAssemblyBaseKernel().
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
default |
Allow instances of this class to be moved.
|
virtualdefault |
Referenced by NEGEMMAssemblyBaseKernel::NEGEMMAssemblyBaseKernel().
|
inline |
Initialise the kernel's input and output.
The computed function is C = a * AxB + b * C.
[in] | input0 | Input tensor containing the Matrix A. Data types supported: F32 |
[in] | input1 | Input tensor containing the Matrix B. Data types supported: same as input0 |
[in,out] | output | Output tensor to store the result of matrix multiplication. If beta is not zero the values are multiplied by beta before the result is accumulated. Otherwise the values are overwritten by the result. Data types supported: same as input0 . |
[out] | workspace | Space for intermediate results. |
[in] | alpha | Weight of the matrix product |
[in] | beta | Weight of the accumulation. |
[in] | is_transposed_0 | (Optional)True if input0 is transposed else false. (Defaults to false) |
[in] | is_transposed_1 | (Optional)True if input1 is transposed else false. (Defaults to false) |
Definition at line 71 of file NEGEMMAssemblyBaseKernel.h.
References arm_compute::test::validation::alpha, and arm_compute::test::validation::beta.
|
inlineoverridevirtual |
Name of the kernel.
Implements ICPPKernel.
Definition at line 37 of file NEGEMMAssemblyBaseKernel.h.
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
Referenced by NEGEMMAssemblyBaseKernel::NEGEMMAssemblyBaseKernel().
|
default |
Allow instances of this class to be moved.