21.02
|
StridedSlice Layer. More...
#include <Layers.h>
Public Member Functions | |
StridedSliceLayer (Coordinates &starts, Coordinates &ends, BiStrides &strides, StridedSliceLayerInfo strided_slice_info) | |
Construct a strided slice 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 strided slice layer.
[in] | starts | The starts of the dimensions of the input tensor to be sliced. The length must be of rank(input). |
[in] | ends | The ends of the dimensions of the input tensor to be sliced. The length must be of rank(input). |
[in] | strides | The strides of the dimensions of the input tensor to be sliced. The length must be of rank(input). |
[in] | strided_slice_info | Contains masks for the starts, ends and strides |
Create layer and add to the given stream.
[in] | s | Stream to add layer to. |
Implements ILayer.
Definition at line 1448 of file Layers.h.
References GraphBuilder::add_strided_slice_node(), IStream::graph(), IStream::hints(), arm_compute::test::validation::input, ILayer::name(), IStream::tail_node(), and StreamHints::target_hint.