24 #ifndef ARM_COMPUTE_GRAPH_REDUCTION_LAYER_NODE_H 25 #define ARM_COMPUTE_GRAPH_REDUCTION_LAYER_NODE_H 48 unsigned int axis()
const;
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
ReductionOperation
Available reduction operations.
ReductionOperation op() const
op accessor
Copyright (c) 2017-2021 Arm Limited.
unsigned int axis() const
axis accessor
bool keep_dims() const
keep_dims accessor
NodeType type() const override
Returns node's type.
ReductionLayerNode(ReductionOperation op, unsigned int axis, bool keep_dims=true)
Default Constructor.
Reduction Operation node.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
void accept(INodeVisitor &v) override
Accepts a node visitor.