diff options
Diffstat (limited to 'arm_compute/graph/nodes/ConcatenateLayerNode.h')
| -rw-r--r-- | arm_compute/graph/nodes/ConcatenateLayerNode.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/arm_compute/graph/nodes/ConcatenateLayerNode.h b/arm_compute/graph/nodes/ConcatenateLayerNode.h index 8582403355..13398b1a61 100644 --- a/arm_compute/graph/nodes/ConcatenateLayerNode.h +++ b/arm_compute/graph/nodes/ConcatenateLayerNode.h @@ -47,7 +47,8 @@ public: * * @return Expected output descriptor */ - static TensorDescriptor compute_output_descriptor(const std::vector<TensorDescriptor> &input_descriptors, DataLayoutDimension axis); + static TensorDescriptor compute_output_descriptor(const std::vector<TensorDescriptor> &input_descriptors, + DataLayoutDimension axis); /** Disables or not the depth concatenate node * * @warning This is used when concatenate is performed using sub-tensors, where this node is used as a placeholder. @@ -78,7 +79,7 @@ public: NodeType type() const override; bool forward_descriptors() override; TensorDescriptor configure_output(size_t idx) const override; - void accept(INodeVisitor &v) override; + void accept(INodeVisitor &v) override; private: unsigned int _total_nodes; |