From e2220551b7a64b929650ba9a60529c31e70c13c5 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Fri, 20 Jul 2018 13:23:44 +0100 Subject: COMPMID-1367: Enable NHWC in graph examples Change-Id: Iabc54a3a1bdcd46a9a921cda39c7c85fef672b72 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/141449 Reviewed-by: Giorgio Arena Reviewed-by: Anthony Barbier Tested-by: Jenkins --- arm_compute/graph/INodeVisitor.h | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) (limited to 'arm_compute/graph/INodeVisitor.h') diff --git a/arm_compute/graph/INodeVisitor.h b/arm_compute/graph/INodeVisitor.h index b5446c4a55..ad390ad760 100644 --- a/arm_compute/graph/INodeVisitor.h +++ b/arm_compute/graph/INodeVisitor.h @@ -51,6 +51,11 @@ public: * @param[in] n Node to visit. */ virtual void visit(BatchNormalizationLayerNode &n) = 0; + /** Visit ConcatenateLayerNode. + * + * @param[in] n Node to visit. + */ + virtual void visit(ConcatenateLayerNode &n) = 0; /** Visit ConstNode. * * @param[in] n Node to visit. @@ -61,11 +66,6 @@ public: * @param[in] n Node to visit. */ virtual void visit(ConvolutionLayerNode &n) = 0; - /** Visit DepthConcatenateLayerNode. - * - * @param[in] n Node to visit. - */ - virtual void visit(DepthConcatenateLayerNode &n) = 0; /** Visit DepthwiseConvolutionLayerNode. * * @param[in] n Node to visit. @@ -148,15 +148,15 @@ public: { default_visit(); } - virtual void visit(ConstNode &n) override + virtual void visit(ConcatenateLayerNode &n) override { default_visit(); } - virtual void visit(ConvolutionLayerNode &n) override + virtual void visit(ConstNode &n) override { default_visit(); } - virtual void visit(DepthConcatenateLayerNode &n) override + virtual void visit(ConvolutionLayerNode &n) override { default_visit(); } -- cgit v1.2.1