Compute Library
 22.05
ROIAlignLayer Class Referencefinal

ROIAlign Layer. More...

#include <Layers.h>

Collaboration diagram for ROIAlignLayer:
[legend]

Public Member Functions

 ROIAlignLayer (SubStream &&sub_stream_input, SubStream &&sub_stream_rois, ROIPoolingLayerInfo pool_info)
 Construct a RoiAlign layer. More...
 
 ROIAlignLayer (const ROIAlignLayer &)=delete
 Prevent instances of this class from being copy constructed. More...
 
ROIAlignLayeroperator= (const ROIAlignLayer &)=delete
 Prevent instances of this class from being copied. 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

ROIAlign Layer.

Definition at line 1236 of file Layers.h.

Constructor & Destructor Documentation

◆ ROIAlignLayer() [1/2]

ROIAlignLayer ( SubStream &&  sub_stream_input,
SubStream &&  sub_stream_rois,
ROIPoolingLayerInfo  pool_info 
)
inline

Construct a RoiAlign layer.

Parameters
[in]sub_stream_inputGraph sub-stream for the input
[in]sub_stream_roisGraph sub-stream for the rois
[in]pool_infoPooling information.

Definition at line 1245 of file Layers.h.

1246  : _ss_input(sub_stream_input), _ss_rois(sub_stream_rois), _pool_info(pool_info)
1247  {
1248  }

◆ ROIAlignLayer() [2/2]

ROIAlignLayer ( const ROIAlignLayer )
delete

Prevent instances of this class from being copy constructed.

Member Function Documentation

◆ create_layer()

NodeID create_layer ( IStream s)
inlineoverridevirtual

Create layer and add to the given stream.

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

Implements ILayer.

Definition at line 1255 of file Layers.h.

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

1256  {
1257  NodeParams common_params = { name(), s.hints().target_hint };
1258  NodeIdxPair input = { _ss_input.tail_node(), 0 };
1259  NodeIdxPair rois = { _ss_rois.tail_node(), 0 };
1260  return GraphBuilder::add_roi_align_node(s.graph(), common_params, input, rois, _pool_info);
1261  }
NodeID tail_node()
Returns the tail node of the Stream.
Definition: IStream.h:65
static NodeID add_roi_align_node(Graph &g, NodeParams params, NodeIdxPair input, NodeIdxPair rois, ROIPoolingLayerInfo pool_info)
Adds a ROI align layer node to the graph.
const std::string & name() const
Layer name accessor.
Definition: ILayer.h:64

◆ operator=()

ROIAlignLayer& operator= ( const ROIAlignLayer )
delete

Prevent instances of this class from being copied.


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