24.02.1
|
Fully Connected Layer. More...
#include <Layers.h>
Public Member Functions | |
FullyConnectedLayer (unsigned int num_outputs, ITensorAccessorUPtr weights, ITensorAccessorUPtr bias, const FullyConnectedLayerInfo fc_info=FullyConnectedLayerInfo(), const QuantizationInfo weights_quant_info=QuantizationInfo(), const QuantizationInfo out_quant_info=QuantizationInfo()) | |
Construct a fully connected layer. More... | |
FullyConnectedLayer (unsigned int num_outputs, SubStream sub_stream_weights, SubStream sub_stream_bias, const FullyConnectedLayerInfo fc_info=FullyConnectedLayerInfo(), const QuantizationInfo weights_quant_info=QuantizationInfo(), const QuantizationInfo out_quant_info=QuantizationInfo()) | |
Construct a fully connected layer. More... | |
NodeID | create_layer (IStream &s) override |
Create layer and add to the given stream. More... | |
Public Member Functions inherited from ILayer | |
virtual | ~ILayer ()=default |
Default destructor. More... | |
ILayer & | set_name (std::string name) |
Sets the name of the layer. More... | |
const std::string & | name () const |
Layer name accessor. More... | |
|
inline |
Construct a fully connected layer.
[in] | num_outputs | Number of outputs. |
[in] | weights | Accessor to get weights from. |
[in] | bias | Accessor to get bias from. |
[in] | fc_info | (Optional) Fully connected layer metadata |
[in] | weights_quant_info | (Optional) Weights quantization information |
[in] | out_quant_info | (Optional) Output quantization info |
Definition at line 726 of file Layers.h.
|
inline |
Construct a fully connected layer.
[in] | num_outputs | Number of outputs. |
[in] | sub_stream_weights | Graph sub-stream for the weights. |
[in] | sub_stream_bias | Graph sub-stream for the bias. |
[in] | fc_info | (Optional) Fully connected layer metadata |
[in] | weights_quant_info | (Optional) Weights quantization information |
[in] | out_quant_info | (Optional) Output quantization info |
Definition at line 752 of file Layers.h.
Create layer and add to the given stream.
[in] | s | Stream to add layer to. |
Implements ILayer.
Definition at line 775 of file Layers.h.
References GraphBuilder::add_fully_connected_layer(), ARM_COMPUTE_ERROR_ON, arm_compute::graph::EmptyNodeID, StreamHints::fast_math_hint, IStream::graph(), IStream::hints(), arm_compute::test::validation::input, ILayer::name(), IStream::tail_node(), and StreamHints::target_hint.