36 : _op(op), _axis(axis), _keep_dims(keep_dims)
TensorShape shape
Tensor shape.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
ReductionOperation
Available reduction operations.
ReductionOperation op() const
op accessor
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
TensorDescriptor & desc()
TensorInfo metadata accessor.
unsigned int axis() const
axis accessor
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
bool keep_dims() const
keep_dims accessor
virtual void visit(INode &n)=0
Visit INode.
Tensor * output(size_t idx) const
Returns the tensor of a given output of the node.
NodeType type() const override
Returns node's type.
TensorShape compute_reduced_shape(const TensorShape &input, unsigned int axis, bool keep_dims=true)
Calculate the reduced shape of a tensor given an axis.
TensorID input_id(size_t idx) const
Returns the tensor ID of a given input of the node.
constexpr EdgeID EmptyEdgeID
ReductionLayerNode(ReductionOperation op, unsigned int axis, bool keep_dims=true)
Default Constructor.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
TensorDescriptor & set_shape(TensorShape &tensor_shape)
Sets tensor descriptor shape.
TensorID output_id(size_t idx) const
Returns the tensor ID of a given output of the node.
constexpr TensorID NullTensorID
Constant NodeID specifying an equivalent of null node.
Tensor * input(size_t idx) const
Returns the tensor of a given input of the node.
void accept(INodeVisitor &v) override
Accepts a node visitor.