24.02.1
|
Go to the documentation of this file.
56 const unsigned int num_tensors = input_descriptors.size();
60 return output_descriptor;
81 bool are_all_inputs_set = std::all_of(std::begin(_input_edges),
std::end(_input_edges),
86 if (are_all_inputs_set)
88 std::vector<TensorDescriptor> inputs_descriptors;
89 for (
unsigned int i = 0; i < _input_edges.size(); ++i)
93 inputs_descriptors.push_back(
t->desc());
TensorShape compute_stack_shape(const ITensorInfo &a, unsigned int axis, unsigned int num_tensors)
Calculate the stack output shape of a tensor.
constexpr EdgeID EmptyEdgeID
TensorID input_id(size_t idx) const
Returns the tensor ID of a given input of the node.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
StackLayerNode(unsigned int total_nodes, int axis)
Constructor.
Tensor * output(size_t idx) const
Returns the tensor of a given output of the node.
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
static TensorDescriptor compute_output_descriptor(const std::vector< TensorDescriptor > &input_descriptors, int axis)
Computes stack output descriptor.
void accept(INodeVisitor &v) override
Accepts a node visitor.
virtual void visit(INode &n)=0
Visit INode.
TensorShape shape
Tensor shape.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
Store the tensor's metadata.
int axis() const
Stack axis parameter accessor.
NodeType type() const override
Returns node's type.
Copyright (c) 2017-2024 Arm Limited.
void end(TokenStream &in, bool &valid)
constexpr TensorID NullTensorID
Constant NodeID specifying an equivalent of null node.