24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_GRAPH_CONCATENATE_LAYER_NODE_H
25 #define ARM_COMPUTE_GRAPH_CONCATENATE_LAYER_NODE_H
85 unsigned int _total_nodes;
void set_enabled(bool is_enabled)
Disables or not the depth concatenate node.
NodeType type() const override
Returns node's type.
Quantization information.
DataLayoutDimension
[DataLayout enum definition]
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
static TensorDescriptor compute_output_descriptor(const std::vector< TensorDescriptor > &input_descriptors, DataLayoutDimension axis)
Computes concatenations output descriptor.
ConcatenateLayerNode(unsigned int total_nodes, descriptors::ConcatLayerDescriptor concat_descriptor)
Constructor.
Concatenation Layer node.
Concatenate layer descriptor.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
QuantizationInfo output_quantization_info() const
Concatenation output quantization info accessor.
DataLayoutDimension concatenation_axis() const
Concatenation axis parameter accessor.
Copyright (c) 2017-2024 Arm Limited.
bool is_enabled() const
Enabled parameter accessor.
void accept(INodeVisitor &v) override
Accepts a node visitor.