21.02
|
Concat Layer. More...
#include <Layers.h>
Public Member Functions | |
template<typename... Ts> | |
ConcatLayer (SubStream &&sub_stream1, SubStream &&sub_stream2, Ts &&... rest_sub_streams) | |
Construct a concatenation layer. More... | |
template<typename... Ts> | |
ConcatLayer (descriptors::ConcatLayerDescriptor concat_descriptor, SubStream &&sub_stream1, SubStream &&sub_stream2, Ts &&... rest_sub_streams) | |
Construct a concatenation layer. More... | |
template<typename... Ts> | |
ConcatLayer (SubStream &&sub_stream) | |
Construct a concat 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 concatenation layer.
[in] | sub_stream1 | First graph branch |
[in] | sub_stream2 | Second graph branch |
[in] | rest_sub_streams | Rest sub-graph branches |
Definition at line 300 of file Layers.h.
References arm_compute::CHANNEL, and arm_compute::utility::for_each().
|
inline |
Construct a concatenation layer.
[in] | concat_descriptor | Concat layer descriptor |
[in] | sub_stream1 | First graph branch |
[in] | sub_stream2 | Second graph branch |
[in] | rest_sub_streams | Rest sub-graph branches |
Definition at line 320 of file Layers.h.
References arm_compute::utility::for_each().
|
inline |
Construct a concat layer.
[in] | sub_stream | Sub-stream |
Definition at line 337 of file Layers.h.
References arm_compute::CHANNEL.
Create layer and add to the given stream.
[in] | s | Stream to add layer to. |
Implements ILayer.
Definition at line 342 of file Layers.h.
References GraphBuilder::add_concatenate_node(), arm_compute::graph::EmptyNodeID, IStream::graph(), IStream::hints(), ILayer::name(), Graph::node(), arm_compute::graph::Output, arm_compute::test::validation::ss(), and StreamHints::target_hint.