24 #ifndef __ARM_COMPUTE_NEFULLYCONNECTEDLAYER_H__ 25 #define __ARM_COMPUTE_NEFULLYCONNECTEDLAYER_H__ 74 FullyConnectedLayerFunctionPtr _run_func;
75 Tensor _weights_transposed;
Base class for all functions.
Basic function to execute GEMM on NEON.
Interface for NEON tensor.
void configure(ITensor *input, ITensor *weights, const ITensor *biases, ITensor *output)
Set the input and output tensors.
NEFullyConnectedLayer()
Constructor.
NEON kernel which transposes the elements of a matrix.
Basic function to simulate a convolution layer.
NEON kernel to add a bias to each row of the input tensor.
Basic implementation of the tensor interface.
Basic function to compute a Fully Connected layer on NEON.
void run() override
Run the kernels contained in the function.