24 #ifndef __ARM_COMPUTE_GRAPH_FULLY_CONNECTED_LAYER_NODE_H__ 25 #define __ARM_COMPUTE_GRAPH_FULLY_CONNECTED_LAYER_NODE_H__ 70 unsigned int _num_outputs;
static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor, unsigned int num_outputs)
Computes fully connected layer output descriptor.
void accept(INodeVisitor &v) override
Accepts a node visitor.
static TensorDescriptor compute_weights_descriptor(const TensorDescriptor &input_descriptor, unsigned int num_outputs)
Computes weights descriptor.
This file contains all available output stages for GEMMLowp on OpenCL.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
FullyConnectedLayerNode(unsigned int num_outputs)
Constructor.
NodeType type() const override
Returns node's type.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
Fully Connected Layer node.
size_t num_outputs() const
Returns number of outputs of the node.