24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_GRAPH_FULLY_CONNECTED_LAYER_NODE_H
25 #define ARM_COMPUTE_GRAPH_FULLY_CONNECTED_LAYER_NODE_H
106 unsigned int _num_outputs;
void set_fused_activation(ActivationLayerInfo fused_activation)
Sets fused activation.
Quantization information.
static TensorDescriptor compute_weights_descriptor(const TensorDescriptor &input_descriptor, unsigned int num_outputs, FullyConnectedLayerInfo fc_info=FullyConnectedLayerInfo(), const QuantizationInfo &weights_quant_info=QuantizationInfo())
Computes weights descriptor.
Fully Connected Layer node.
Activation Layer Information class.
NodeType type() const override
Returns node's type.
Fully connected layer info.
FullyConnectedLayerNode(unsigned int num_outputs, QuantizationInfo out_quant_info=QuantizationInfo(), FullyConnectedLayerInfo fc_info=FullyConnectedLayerInfo(), FastMathHint fast_math_hint=FastMathHint::Disabled)
Constructor.
void set_fast_math_hint(FastMathHint hint)
Sets the fast math fast hint.
FullyConnectedLayerInfo info() const
Fully connected layer addition information.
@ Disabled
Fast math disabled for Convolution layer.
void accept(INodeVisitor &v) override
Accepts a node visitor.
static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor, unsigned int num_outputs, const QuantizationInfo &out_quant_info=QuantizationInfo())
Computes fully connected layer output descriptor.
static constexpr NodeType node_type
FastMathHint
Enable or disable fast math for Convolution layer.
Copyright (c) 2017-2024 Arm Limited.
FastMathHint fast_math_hint() const
Fast math hint accessor.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
size_t num_outputs() const
Returns number of outputs of the node.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.