From acce504ec4aebe5e5da470c1cfc3cee401ff11f3 Mon Sep 17 00:00:00 2001 From: giuros01 Date: Thu, 21 Feb 2019 17:32:34 +0000 Subject: COMPMID-1740: Fuse batch normalization with Convolution Layer at graph level Change-Id: I77ca51c2c72783cc26a099a6a9c3210cdbbe822d Signed-off-by: giuros01 Reviewed-on: https://review.mlplatform.org/c/797 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Reviewed-by: Georgios Pinitas --- arm_compute/graph/INodeVisitor.h | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) (limited to 'arm_compute/graph/INodeVisitor.h') diff --git a/arm_compute/graph/INodeVisitor.h b/arm_compute/graph/INodeVisitor.h index 573d642892..842ca4bfb3 100644 --- a/arm_compute/graph/INodeVisitor.h +++ b/arm_compute/graph/INodeVisitor.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018 ARM Limited. + * Copyright (c) 2018-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -91,6 +91,11 @@ public: * @param[in] n Node to visit. */ virtual void visit(FullyConnectedLayerNode &n) = 0; + /** Visit FusedConvolutionBatchNormalizationNode. + * + * @param[in] n Node to visit. + */ + virtual void visit(FusedConvolutionBatchNormalizationNode &n) = 0; /** Visit InputNode. * * @param[in] n Node to visit. @@ -195,6 +200,10 @@ public: { default_visit(); } + virtual void visit(FusedConvolutionBatchNormalizationNode &n) override + { + default_visit(); + } virtual void visit(InputNode &n) override { default_visit(); -- cgit v1.2.1