24 #ifndef ARM_COMPUTE_GRAPH_STACK_LAYER_NODE_H 25 #define ARM_COMPUTE_GRAPH_STACK_LAYER_NODE_H 64 unsigned int _total_nodes;
static TensorDescriptor compute_output_descriptor(const std::vector< TensorDescriptor > &input_descriptors, int axis)
Computes stack output descriptor.
NodeType type() const override
Returns node's type.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
StackLayerNode(unsigned int total_nodes, int axis)
Constructor.
Copyright (c) 2017-2021 Arm Limited.
int axis() const
Stack axis parameter accessor.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
void accept(INodeVisitor &v) override
Accepts a node visitor.