Compute Library
PriorBoxLayer Class Referencefinal

PriorBox Layer. More...

#include <Layers.h>

Collaboration diagram for PriorBoxLayer:

Public Member Functions

 PriorBoxLayer (SubStream &&sub_stream, const PriorBoxLayerInfo &prior_info)
 Construct a priorbox 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

PriorBox Layer.

Definition at line 1086 of file Layers.h.

Constructor & Destructor Documentation

◆ PriorBoxLayer()

PriorBoxLayer ( SubStream &&  sub_stream,
const PriorBoxLayerInfo prior_info 

Construct a priorbox layer.

[in]sub_streamFirst graph sub-stream
[in]prior_infoPriorBox parameters.

Definition at line 1094 of file Layers.h.

1095  : _ss(std::move(sub_stream)), _prior_info(prior_info)
1096  {
1097  }

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 1099 of file Layers.h.

References GraphBuilder::add_priorbox_node(), IStream::graph(), IStream::hints(), ILayer::name(), IStream::tail_node(), and StreamHints::target_hint.

1100  {
1101  NodeParams common_params = { name(), s.hints().target_hint };
1102  NodeIdxPair input0 = { s.tail_node(), 0 };
1103  NodeIdxPair input1 = { _ss.tail_node(), 0 };
1104  return GraphBuilder::add_priorbox_node(s.graph(), common_params, input0, input1, _prior_info);
1105  }
static NodeID add_priorbox_node(Graph &g, NodeParams params, NodeIdxPair input0, NodeIdxPair input1, const PriorBoxLayerInfo &prior_info)
Adds a priorbox layer node to the graph.
NodeID tail_node()
Returns the tail node of the Stream.
Definition: IStream.h:65
const std::string & name() const
Layer name accessor.
Definition: ILayer.h:64

The documentation for this class was generated from the following file: