Slice Layer node. More...

#include <StridedSliceLayerNode.h>

Collaboration diagram for StridedSliceLayerNode:

Public Member Functions
	StridedSliceLayerNode (const Coordinates &starts, const Coordinates &ends, const BiStrides &strides, StridedSliceLayerInfo strided_slice_info)
	Default Constructor. More...

Coordinates	starts () const
	Start coordinates accessor. More...

Coordinates	ends () const
	End coordinates accessor. More...

BiStrides	strides () const
	Strides vector accessor. More...

StridedSliceLayerInfo	strided_slice_info () const

NodeType	type () const override
	Returns node's type. More...

bool	forward_descriptors () override
	Forwards descriptor information to outputs if possible. More...

TensorDescriptor	configure_output (size_t idx) const override
	Calculates output configuration. More...

void	accept (INodeVisitor &v) override
	Accepts a node visitor. More...

Public Member Functions inherited from INode
	INode ()
	Constructor. More...

virtual	~INode ()=default
	Destructor. More...

	INode (const INode &)=delete
	Prevent instances of this class from being copied (As this class contains pointers) More...

INode &	operator= (const INode &)=delete
	Prevent instances of this class from being copy assigned (As this class contains pointers) More...

	INode (INode &&)=default
	Allow instances of this class to be moved. More...

INode &	operator= (INode &&)=default
	Allow instances of this class to be move assigned. More...

virtual Status	validate () const
	Validate node. More...

std::string	name () const
	Returns node's name. More...

NodeID	id () const
	Returns node's ID. More...

const Graph *	graph () const
	Returns node's Graph. More...

Graph *	graph ()
	Returns node's Graph. More...

void	set_graph (Graph *g)
	Sets the graph that this node is registered to. More...

void	set_id (NodeID id)
	Sets the node id. More...

void	set_common_node_parameters (NodeParams common_params)
	Sets common node parameters. More...

void	set_requested_target (Target target)
	Sets target preference. More...

void	set_assigned_target (Target target)
	Sets the final execution target. More...

void	set_output_tensor (TensorID tid, size_t idx)
	Sets the output tensor of at a given index. More...

const std::vector< TensorID > &	inputs () const
	Returns inputs of the node. More...

const std::vector< TensorID > &	outputs () const
	Returns outputs of the node. More...

const std::vector< EdgeID > &	input_edges () const
	Returns input edge set. More...

const std::set< EdgeID > &	output_edges () const
	Returns output edge set. More...

TensorID	input_id (size_t idx) const
	Returns the tensor ID of a given input of the node. More...

TensorID	output_id (size_t idx) const
	Returns the tensor ID of a given output of the node. More...

Tensor *	input (size_t idx) const
	Returns the tensor of a given input of the node. More...

Tensor *	output (size_t idx) const
	Returns the tensor of a given output of the node. More...

EdgeID	input_edge_id (size_t idx) const
	Returns the edge ID of a given input of the node. More...

Edge *	input_edge (size_t idx) const
	Returns the edge of a given input of the node. More...

size_t	num_inputs () const
	Returns number of inputs of the node. More...

size_t	num_outputs () const
	Returns number of outputs of the node. More...

NodeParams	common_node_params () const
	Returns common node parameters. More...

Target	requested_target () const
	Returns requested target for this node. More...

Target	assigned_target () const
	Returns assigned target for this node. More...

Static Public Member Functions
static TensorDescriptor	compute_output_descriptor (const TensorDescriptor &input_descriptor, const Coordinates &starts, const Coordinates &ends, const BiStrides &strides, StridedSliceLayerInfo info)
	Computes slice layer output descriptor. More...

Detailed Description

Slice Layer node.

Definition at line 36 of file StridedSliceLayerNode.h.

Constructor & Destructor Documentation

◆ StridedSliceLayerNode()

StridedSliceLayerNode	(	const Coordinates &	starts,
		const Coordinates &	ends,
		const BiStrides &	strides,
		StridedSliceLayerInfo	strided_slice_info
	)

Default Constructor.

Parameters

[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

Definition at line 35 of file StridedSliceLayerNode.cpp.

     : _starts(starts), _ends(ends), _strides(strides), _info(std::move(info))
 {
     _input_edges.resize(1, EmptyEdgeID);
     _outputs.resize(1, NullTensorID);
 }

Member Function Documentation

◆ accept()

void accept ( INodeVisitor & v )

overridevirtual

Accepts a node visitor.

Parameters

[in] v Visitor to accept

Implements INode.

Definition at line 108 of file StridedSliceLayerNode.cpp.

References INodeVisitor::visit().

 {
     v.visit(*this);
 }

◆ compute_output_descriptor()

TensorDescriptor compute_output_descriptor	(	const TensorDescriptor &	input_descriptor,
		const Coordinates &	starts,
		const Coordinates &	ends,
		const BiStrides &	strides,
		StridedSliceLayerInfo	info
	)

static

Computes slice layer output descriptor.

Parameters

[in]	input_descriptor	Descriptor of the input tensor
[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]	info	Contains masks for the starts, ends and strides

Returns: Output descriptor

Definition at line 65 of file StridedSliceLayerNode.cpp.

References StridedSliceLayerInfo::begin_mask(), arm_compute::helpers::tensor_transform::compute_strided_slice_output_shape(), StridedSliceLayerInfo::end_mask(), TensorDescriptor::shape, and StridedSliceLayerInfo::shrink_axis_mask().

Referenced by StridedSliceLayerNode::configure_output().

 {
     using namespace arm_compute::helpers::tensor_transform;
 
     TensorDescriptor output_desc = input_descriptor;
     output_desc.shape            = compute_strided_slice_output_shape(input_descriptor.shape, starts, ends, strides,
                                                                       info.begin_mask(), info.end_mask(), info.shrink_axis_mask());
 
     return output_desc;
 }

◆ configure_output()

TensorDescriptor configure_output ( size_t idx ) const

overridevirtual

Calculates output configuration.

Parameters

[in] idx Output index to configure

Returns: Output descriptor configuration

Implements INode.

Definition at line 92 of file StridedSliceLayerNode.cpp.

References ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_UNUSED, StridedSliceLayerNode::compute_output_descriptor(), INode::input(), and arm_compute::test::validation::src.

Referenced by StridedSliceLayerNode::forward_descriptors().

 {
     ARM_COMPUTE_UNUSED(idx);
     ARM_COMPUTE_ERROR_ON(idx >= _outputs.size());
 
     const Tensor *src = input(0);
     ARM_COMPUTE_ERROR_ON(src == nullptr);
 
     return compute_output_descriptor(src->desc(), _starts, _ends, _strides, _info);
 }

◆ ends()

Coordinates ends ( ) const

End coordinates accessor.

Returns: End coordinates of the dimensions

Definition at line 50 of file StridedSliceLayerNode.cpp.

Referenced by arm_compute::graph::backends::detail::validate_strided_slice_layer().

 {
     return _ends;
 }

◆ forward_descriptors()

bool forward_descriptors ( )

overridevirtual

Forwards descriptor information to outputs if possible.

Returns: True if descriptor information could be forwarded otherwise false

Implements INode.

Definition at line 80 of file StridedSliceLayerNode.cpp.

References ARM_COMPUTE_ERROR_ON, StridedSliceLayerNode::configure_output(), Tensor::desc(), arm_compute::test::validation::dst, INode::input_id(), arm_compute::graph::NullTensorID, INode::output(), and INode::output_id().

 {
     if((input_id(0) != NullTensorID) && (output_id(0) != NullTensorID))
     {
         Tensor *dst = output(0);
         ARM_COMPUTE_ERROR_ON(dst == nullptr);
         dst->desc() = configure_output(0);
         return true;
     }
     return false;
 }

◆ starts()

Coordinates starts ( ) const

Start coordinates accessor.

Returns: Start coordinates of the dimensions

Definition at line 45 of file StridedSliceLayerNode.cpp.

Referenced by arm_compute::graph::backends::detail::validate_strided_slice_layer().

 {
     return _starts;
 }

◆ strided_slice_info()

StridedSliceLayerInfo strided_slice_info ( ) const

Definition at line 60 of file StridedSliceLayerNode.cpp.

Referenced by arm_compute::graph::backends::detail::validate_strided_slice_layer().

 {
     return _info;
 }

◆ strides()

BiStrides strides ( ) const

Strides vector accessor.

Returns: End coordinates of the dimensions

Definition at line 55 of file StridedSliceLayerNode.cpp.

Referenced by arm_compute::graph::backends::detail::validate_strided_slice_layer().

 {
     return _strides;
 }

◆ type()

NodeType type ( ) const

overridevirtual

Returns node's type.

Returns: Node's type

Implements INode.

Definition at line 103 of file StridedSliceLayerNode.cpp.

References arm_compute::graph::StridedSliceLayer.

 {
     return NodeType::StridedSliceLayer;
 }

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

arm_compute/graph/nodes/StridedSliceLayerNode.h
src/graph/nodes/StridedSliceLayerNode.cpp

Public Member Functions

Static Public Member Functions

Detailed Description

Constructor & Destructor Documentation

◆ StridedSliceLayerNode()

Member Function Documentation

◆ accept()

◆ compute_output_descriptor()

◆ configure_output()

◆ ends()

◆ forward_descriptors()

◆ starts()

◆ strided_slice_info()

◆ strides()

◆ type()