diff options
Diffstat (limited to 'src/graph/nodes/ConcatenateLayerNode.cpp')
| -rw-r--r-- | src/graph/nodes/ConcatenateLayerNode.cpp | 17 |
1 files changed, 13 insertions, 4 deletions
diff --git a/src/graph/nodes/ConcatenateLayerNode.cpp b/src/graph/nodes/ConcatenateLayerNode.cpp index 3ce09d0073..bbdc4dc029 100644 --- a/src/graph/nodes/ConcatenateLayerNode.cpp +++ b/src/graph/nodes/ConcatenateLayerNode.cpp @@ -34,8 +34,8 @@ namespace arm_compute { namespace graph { -ConcatenateLayerNode::ConcatenateLayerNode(unsigned int total_nodes, DataLayoutDimension axis) - : _total_nodes(total_nodes), _axis(axis), _is_enabled(true) +ConcatenateLayerNode::ConcatenateLayerNode(unsigned int total_nodes, descriptors::ConcatLayerDescriptor concat_descriptor) + : _total_nodes(total_nodes), _concat_descriptor(std::move(concat_descriptor)), _is_enabled(true) { _input_edges.resize(_total_nodes, EmptyEdgeID); _outputs.resize(1, NullTensorID); @@ -53,7 +53,12 @@ bool ConcatenateLayerNode::is_enabled() const DataLayoutDimension ConcatenateLayerNode::concatenation_axis() const { - return _axis; + return _concat_descriptor.axis; +} + +QuantizationInfo ConcatenateLayerNode::output_quantization_info() const +{ + return _concat_descriptor.output_qinfo; } TensorDescriptor ConcatenateLayerNode::compute_output_descriptor(const std::vector<TensorDescriptor> &input_descriptors, @@ -121,7 +126,11 @@ TensorDescriptor ConcatenateLayerNode::configure_output(size_t idx) const ARM_COMPUTE_ERROR_ON(t == nullptr); inputs_descriptors.push_back(t->desc()); } - output_info = compute_output_descriptor(inputs_descriptors, _axis); + output_info = compute_output_descriptor(inputs_descriptors, _concat_descriptor.axis); + if(!_concat_descriptor.output_qinfo.empty()) + { + output_info.quant_info = _concat_descriptor.output_qinfo; + } } return output_info; |