24 #ifndef __ARM_COMPUTE_NELOCALLYCONNECTEDMATRIXMULTIPLYKERNEL_H__ 25 #define __ARM_COMPUTE_NELOCALLYCONNECTEDMATRIXMULTIPLYKERNEL_H__ 37 const char *
name()
const override 39 return "NELocallyConnectedMatrixMultiplyKernel";
Common interface for all kernels implemented in C++.
Store the tensor's metadata.
Interface for NEON tensor.
src info() -> set_format(Format::S16)
This file contains all available output stages for GEMMLowp on OpenCL.
void run(const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
NELocallyConnectedMatrixMultiplyKernel & operator=(const NELocallyConnectedMatrixMultiplyKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
NELocallyConnectedMatrixMultiplyKernel()
Default constructor.
NEON kernel to multiply each row of first tensor with low 2 dimensions of second tensor.
static Status validate(const ITensorInfo *input0, const ITensorInfo *input1, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of NELocallyConnectedMatrix...
void configure(const ITensor *input0, const ITensor *input1, ITensor *output)
Initialise the kernel's input and output.
const char * name() const override
Name of the kernel.
Information about executing thread and CPU.
const Window & window() const
The maximum window the kernel can be executed on.
Describe a multidimensional execution window.