24.02.1
|
Go to the documentation of this file.
37 : _num_outputs(num_outputs),
38 _out_quant_info(std::move(out_quant_info)),
40 _fast_math_hint(fast_math_hint)
47 _fast_math_hint = hint;
52 return _fast_math_hint;
61 unsigned int num_outputs,
65 unsigned int num_weights = 1;
68 if (num_dimensions == 2 || num_dimensions == 4)
72 for (
unsigned int i = 0; i < num_dimensions; i++)
74 num_weights *= input_descriptor.
shape[i];
87 if (!weights_quant_info.
empty())
89 weights_descriptor.
quant_info = weights_quant_info;
92 return weights_descriptor;
96 unsigned int num_outputs,
111 if (!out_quant_info.
empty())
113 output_descriptor.
quant_info = out_quant_info;
116 return output_descriptor;
constexpr EdgeID EmptyEdgeID
SimpleTensor< float > src
void set_fused_activation(ActivationLayerInfo fused_activation)
Sets fused activation.
Quantization information.
TensorID input_id(size_t idx) const
Returns the tensor ID of a given input of the node.
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.
Tensor * output(size_t idx) const
Returns the tensor of a given output of the node.
QuantizationInfo quant_info
Quantization info.
ActivationLayerInfo activation_info
Fused activation to apply after the matrix multiplication.
Activation Layer Information class.
NodeType type() const override
Returns node's type.
Fully connected layer info.
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
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.
virtual void visit(INode &n)=0
Visit INode.
void accept(INodeVisitor &v) override
Accepts a node visitor.
TensorShape shape
Tensor shape.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
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.
FastMathHint
Enable or disable fast math for Convolution layer.
Copyright (c) 2017-2024 Arm Limited.
bool empty() const
Indicates whether this QuantizationInfo has valid settings or not.
FastMathHint fast_math_hint() const
Fast math hint accessor.
Tensor * input(size_t idx) const
Returns the tensor of a given input of the node.
TensorID output_id(size_t idx) const
Returns the tensor ID of a given output of the node.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
size_t num_outputs() const
Returns number of outputs of the node.
constexpr TensorID NullTensorID
Constant NodeID specifying an equivalent of null node.
bool transpose_weights
Transpose weights if true.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
unsigned int num_dimensions() const
Returns the effective dimensionality of the tensor.