24.02
|
Stream frontend class to construct simple graphs in a stream fashion. More...
#include <Stream.h>
Public Member Functions | |
Stream (size_t id, std::string name) | |
Constructor. More... | |
Stream (const Stream &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
Stream & | operator= (const Stream &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
void | finalize (Target target, const GraphConfig &config) |
Finalizes the stream for an execution target. More... | |
void | run () |
Executes the stream. More... | |
void | add_layer (ILayer &layer) override |
Adds a layer to the stream. More... | |
Graph & | graph () override |
Returns the underlying graph. More... | |
const Graph & | graph () const override |
Returns the underlying graph. More... | |
Public Member Functions inherited from IStream | |
virtual | ~IStream ()=default |
NodeID | tail_node () |
Returns the tail node of the Stream. More... | |
StreamHints & | hints () |
Returns the stream hints that are currently used. More... | |
void | forward_tail (NodeID nid) |
Forwards tail of stream to a given nid. More... | |
Stream frontend class to construct simple graphs in a stream fashion.
Stream | ( | size_t | id, |
std::string | name | ||
) |
Constructor.
Definition at line 35 of file Stream.cpp.
Prevent instances of this class from being copied (As this class contains pointers)
|
overridevirtual |
Adds a layer to the stream.
[in] | layer | Layer to add |
Implements IStream.
Definition at line 51 of file Stream.cpp.
References ILayer::create_layer().
void finalize | ( | Target | target, |
const GraphConfig & | config | ||
) |
Finalizes the stream for an execution target.
[in] | target | Execution target |
[in] | config | (Optional) Graph configuration to use |
Definition at line 39 of file Stream.cpp.
References arm_compute::graph::create_default_pass_manager(), GraphManager::finalize_graph(), and GraphContext::set_config().
Referenced by GraphValidateExample< DepthwiseConvolutionLayer, DepthConvolutionOptions, DepthConvolutionVerifyAccessor >::do_setup().
|
overridevirtual |
|
overridevirtual |
Returns the underlying graph.
Implements IStream.
Definition at line 57 of file Stream.cpp.
Prevent instances of this class from being copied (As this class contains pointers)
void run | ( | ) |
Executes the stream.
Definition at line 46 of file Stream.cpp.
References GraphManager::execute_graph().
Referenced by GraphValidateExample< DepthwiseConvolutionLayer, DepthConvolutionOptions, DepthConvolutionVerifyAccessor >::do_run().