24 #ifndef __ARM_COMPUTE_NEFULLYCONNECTEDLAYER_H__ 25 #define __ARM_COMPUTE_NEFULLYCONNECTEDLAYER_H__ 79 bool _transpose_weights;
80 bool _is_batched_fc_layer;
139 Tensor _interleave4x4_output;
140 Tensor _reshape_weights_output;
141 bool _are_weights_reshaped;
142 bool _is_batched_fc_layer;
143 bool _linearize_input;
144 bool _accumulate_biases;
145 const ITensor *_original_weights;
Base class for all functions.
Basic function to reshape the weights of Fully Connected layer with NEON.
void configure(const ITensor *input, ITensor *output, bool transpose_weights, bool is_batched_fc_layer)
Set the input and output tensors.
Interface for the im2col reshape kernel.
Store the tensor's metadata.
Interface for NEON tensor.
This file contains all available output stages for GEMMLowp on OpenCL.
NEON kernel to interleave the elements of a matrix.
NEON kernel which transposes the elements of a matrix.
NEON kernel to add a bias to each row of the input tensor.
NEON kernel which transposes the elements of a matrix in chunks of 1xW, where W is equal to (16 / ele...
Basic implementation of the tensor interface.
void run() override
Run the kernels contained in the function.
Basic function to compute a Fully Connected layer on NEON.
NEFullyConnectedLayerReshapeWeights(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Constructor.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, bool transpose_weights, bool is_batched_fc_layer)
Static function to check if given info will lead to a valid configuration of CLFullyConnectedLayerRes...
NEON kernel to multiply two input matrices "A" and "B".