ReductionLayerNode Class Reference

Reduction Operation node. More...

#include <ReductionLayerNode.h>

Collaboration diagram for ReductionLayerNode:

Public Member Functions
	ReductionLayerNode (ReductionOperation op, unsigned int axis, bool keep_dims=true)
	Default Constructor. More...
ReductionOperation	op () const
	op accessor More...
unsigned int	axis () const
	axis accessor More...
bool	keep_dims () const
	keep_dims accessor 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

Reduction Operation node.

Definition at line 34 of file ReductionLayerNode.h.

Constructor & Destructor Documentation

ReductionLayerNode()

ReductionLayerNode	(	ReductionOperation	op,
		unsigned int	axis,
		bool	keep_dims = true
	)

Default Constructor.

Definition at line 35 of file ReductionLayerNode.cpp.

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

    : _op(op), _axis(axis), _keep_dims(keep_dims)
{
    _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 89 of file ReductionLayerNode.cpp.

References INodeVisitor::visit().

{
    v.visit(*this);
}

axis()

unsigned int axis

(

)

const

axis accessor

Returns

axis

Definition at line 47 of file ReductionLayerNode.cpp.

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

{
    return _axis;
}

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 69 of file ReductionLayerNode.cpp.

References ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_UNUSED, arm_compute::misc::shape_calculator::compute_reduced_shape(), INode::input(), arm_compute::test::validation::output_info, arm_compute::test::validation::output_shape, TensorDescriptor::set_shape(), TensorDescriptor::shape, and arm_compute::test::validation::src.

Referenced by ReductionLayerNode::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_info  = src->desc();
    TensorShape      output_shape = arm_compute::misc::shape_calculator::compute_reduced_shape(output_info.shape, _axis, _keep_dims);
    output_info.set_shape(output_shape);

    return output_info;
}

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 57 of file ReductionLayerNode.cpp.

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

keep_dims()

bool keep_dims

(

)

const

keep_dims accessor

Returns

keep_dims

Definition at line 52 of file ReductionLayerNode.cpp.

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

{
    return _keep_dims;
}

op()

ReductionOperation op

(

)

const

op accessor

Returns

op

Definition at line 42 of file ReductionLayerNode.cpp.

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

{
    return _op;
}

type()

NodeType type ( ) const

overridevirtual

Returns node's type.

Returns

Node's type

Implements INode.

Definition at line 84 of file ReductionLayerNode.cpp.

References arm_compute::graph::ReductionOperationLayer.

{
    return NodeType::ReductionOperationLayer;
}

---

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

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