24.02.1
|
Go to the documentation of this file.
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.
Reduction Operation node.
ReductionOperation op() const
op accessor
ReductionOperation
Available reduction operations.
bool keep_dims() const
keep_dims accessor
unsigned int axis() const
axis accessor
NodeType type() const override
Returns node's type.
ReductionLayerNode(ReductionOperation op, unsigned int axis, bool keep_dims=true)
Default Constructor.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
Copyright (c) 2017-2024 Arm Limited.
void accept(INodeVisitor &v) override
Accepts a node visitor.