24.02
|
#include <IStream.h>
Public Member Functions | |
virtual | ~IStream ()=default |
virtual void | add_layer (ILayer &layer)=0 |
Adds a layer to the stream. More... | |
virtual Graph & | graph ()=0 |
Returns the underlying graph. More... | |
virtual const Graph & | graph () const =0 |
Returns the underlying graph. More... | |
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... | |
|
virtualdefault |
|
pure virtual |
Adds a layer to the stream.
[in] | layer | Layer to add |
Implemented in Stream, and SubStream.
Referenced by arm_compute::graph::frontend::operator<<().
|
inline |
Forwards tail of stream to a given nid.
[in] | nid | NodeID of the updated tail node |
Definition at line 81 of file IStream.h.
References arm_compute::graph::NullTensorID.
|
pure virtual |
|
pure virtual |
Returns the underlying graph.
Implemented in Stream, and SubStream.
Referenced by InputLayer::create_layer(), ConstantLayer::create_layer(), OutputLayer::create_layer(), ActivationLayer::create_layer(), ArgMinMaxLayer::create_layer(), BatchNormalizationLayer::create_layer(), BoundingBoxTransformLayer::create_layer(), ChannelShuffleLayer::create_layer(), ConcatLayer::create_layer(), ConvolutionLayer::create_layer(), DeconvolutionLayer::create_layer(), DepthwiseConvolutionLayer::create_layer(), DepthToSpaceLayer::create_layer(), DequantizationLayer::create_layer(), DetectionOutputLayer::create_layer(), DetectionPostProcessLayer::create_layer(), DummyLayer::create_layer(), EltwiseLayer::create_layer(), FlattenLayer::create_layer(), FullyConnectedLayer::create_layer(), GenerateProposalsLayer::create_layer(), L2NormalizeLayer::create_layer(), NormalizationLayer::create_layer(), NormalizePlanarYUVLayer::create_layer(), PadLayer::create_layer(), PermuteLayer::create_layer(), PoolingLayer::create_layer(), PReluLayer::create_layer(), PrintLayer::create_layer(), PriorBoxLayer::create_layer(), QuantizationLayer::create_layer(), ReductionLayer::create_layer(), ReorgLayer::create_layer(), ReshapeLayer::create_layer(), ResizeLayer::create_layer(), ROIAlignLayer::create_layer(), ScaleLayer::create_layer(), SliceLayer::create_layer(), SoftmaxLayer::create_layer(), StackLayer::create_layer(), StridedSliceLayer::create_layer(), YOLOLayer::create_layer(), and SubStream::graph().
|
inline |
Returns the stream hints that are currently used.
Definition at line 73 of file IStream.h.
Referenced by InputLayer::create_layer(), ConstantLayer::create_layer(), OutputLayer::create_layer(), ActivationLayer::create_layer(), ArgMinMaxLayer::create_layer(), BatchNormalizationLayer::create_layer(), BoundingBoxTransformLayer::create_layer(), ChannelShuffleLayer::create_layer(), ConcatLayer::create_layer(), ConvolutionLayer::create_layer(), DeconvolutionLayer::create_layer(), DepthwiseConvolutionLayer::create_layer(), DepthToSpaceLayer::create_layer(), DequantizationLayer::create_layer(), DetectionOutputLayer::create_layer(), DetectionPostProcessLayer::create_layer(), DummyLayer::create_layer(), EltwiseLayer::create_layer(), FlattenLayer::create_layer(), FullyConnectedLayer::create_layer(), GenerateProposalsLayer::create_layer(), L2NormalizeLayer::create_layer(), NormalizationLayer::create_layer(), NormalizePlanarYUVLayer::create_layer(), PadLayer::create_layer(), PermuteLayer::create_layer(), PoolingLayer::create_layer(), PReluLayer::create_layer(), PrintLayer::create_layer(), PriorBoxLayer::create_layer(), QuantizationLayer::create_layer(), ReductionLayer::create_layer(), ReorgLayer::create_layer(), ReshapeLayer::create_layer(), ResizeLayer::create_layer(), ROIAlignLayer::create_layer(), ScaleLayer::create_layer(), SliceLayer::create_layer(), SoftmaxLayer::create_layer(), StackLayer::create_layer(), StridedSliceLayer::create_layer(), YOLOLayer::create_layer(), arm_compute::graph::frontend::operator<<(), and SubStream::SubStream().
|
inline |
Returns the tail node of the Stream.
Definition at line 65 of file IStream.h.
Referenced by OutputLayer::create_layer(), ActivationLayer::create_layer(), ArgMinMaxLayer::create_layer(), BatchNormalizationLayer::create_layer(), BoundingBoxTransformLayer::create_layer(), ChannelShuffleLayer::create_layer(), ConvolutionLayer::create_layer(), DeconvolutionLayer::create_layer(), DepthwiseConvolutionLayer::create_layer(), DepthToSpaceLayer::create_layer(), DequantizationLayer::create_layer(), DetectionOutputLayer::create_layer(), DetectionPostProcessLayer::create_layer(), DummyLayer::create_layer(), EltwiseLayer::create_layer(), FlattenLayer::create_layer(), FullyConnectedLayer::create_layer(), GenerateProposalsLayer::create_layer(), L2NormalizeLayer::create_layer(), NormalizationLayer::create_layer(), NormalizePlanarYUVLayer::create_layer(), PadLayer::create_layer(), PermuteLayer::create_layer(), PoolingLayer::create_layer(), PReluLayer::create_layer(), PrintLayer::create_layer(), PriorBoxLayer::create_layer(), QuantizationLayer::create_layer(), ReductionLayer::create_layer(), ReorgLayer::create_layer(), ReshapeLayer::create_layer(), ResizeLayer::create_layer(), ROIAlignLayer::create_layer(), ScaleLayer::create_layer(), SliceLayer::create_layer(), SoftmaxLayer::create_layer(), StridedSliceLayer::create_layer(), YOLOLayer::create_layer(), and SubStream::SubStream().