Compute Library
OutputLayer Class Referencefinal

Output Layer. More...

#include <Layers.h>

Collaboration diagram for OutputLayer:

Public Member Functions

 OutputLayer (ITensorAccessorUPtr accessor, unsigned int connection_idx=0)
 Construct an output 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...
ILayerset_name (std::string name)
 Sets the name of the layer. More...
const std::string & name () const
 Layer name accessor. More...

Detailed Description

Output Layer.

Definition at line 92 of file Layers.h.

Constructor & Destructor Documentation

◆ OutputLayer()

OutputLayer ( ITensorAccessorUPtr  accessor,
unsigned int  connection_idx = 0 

Construct an output layer.

[in]accessorAccessor to give output tensor data to.
[in]connection_idx(Optional) Input connection index

Definition at line 100 of file Layers.h.

101  : _accessor(std::move(accessor)), _connection_idx(connection_idx)
102  {
103  }

Member Function Documentation

◆ create_layer()

NodeID create_layer ( IStream s)

Create layer and add to the given stream.

[in]sStream to add layer to.
ID of the created node.

Implements ILayer.

Definition at line 105 of file Layers.h.

106  {
107  NodeParams common_params = {name(), s.hints().target_hint};
108  NodeIdxPair input = {s.tail_node(), _connection_idx};
109  return GraphBuilder::add_output_node(s.graph(), common_params, input, std::move(_accessor));
110  }

References GraphBuilder::add_output_node(), IStream::graph(), IStream::hints(), arm_compute::test::validation::input, ILayer::name(), IStream::tail_node(), and StreamHints::target_hint.

The documentation for this class was generated from the following file:
const std::string & name() const
Layer name accessor.
Definition: ILayer.h:64
static NodeID add_output_node(Graph &g, NodeParams params, NodeIdxPair input, ITensorAccessorUPtr accessor=nullptr)
Adds an output layer node to the graph.
Definition: GraphBuilder.cpp:132
auto input
Definition: LSTMLayerQuantized.cpp:486