FlattenLayerNode Class Reference

Flatten Layer node. More...

#include <FlattenLayerNode.h>

Collaboration diagram for FlattenLayerNode:

Public Member Functions
	FlattenLayerNode ()
	Default Constructor. More...
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...

Detailed Description

Flatten Layer node.

Definition at line 34 of file FlattenLayerNode.h.

Constructor & Destructor Documentation

FlattenLayerNode()

FlattenLayerNode

(

)

Default Constructor.

Definition at line 33 of file FlattenLayerNode.cpp.

References arm_compute::graph::EmptyEdgeID, and arm_compute::graph::NullTensorID.

{
    _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 70 of file FlattenLayerNode.cpp.

References INodeVisitor::visit().

{
    v.visit(*this);
}

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 51 of file FlattenLayerNode.cpp.

References ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_UNUSED, TensorShape::collapse(), INode::input(), TensorDescriptor::shape, and arm_compute::test::validation::src.

Referenced by FlattenLayerNode::forward_descriptors().

{
    ARM_COMPUTE_UNUSED(idx);
    ARM_COMPUTE_ERROR_ON(idx >= _outputs.size());

    const Tensor *src = input(0);
    ARM_COMPUTE_ERROR_ON(src == nullptr);

    TensorDescriptor output_desc = src->desc();
    output_desc.shape.collapse(3);

    return output_desc;
}

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 39 of file FlattenLayerNode.cpp.

References ARM_COMPUTE_ERROR_ON, FlattenLayerNode::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;
}

type()

NodeType type ( ) const

overridevirtual

Returns node's type.

Returns

Node's type

Implements INode.

Definition at line 65 of file FlattenLayerNode.cpp.

References arm_compute::graph::FlattenLayer.

{
    return NodeType::FlattenLayer;
}

---

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

• arm_compute/graph/nodes/FlattenLayerNode.h
• src/graph/nodes/FlattenLayerNode.cpp