From 1c32bf396eb690a54fd94487e3f258b2c7d31753 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Mon, 12 Nov 2018 18:36:19 +0000 Subject: COMPMID-1451: Perform fusion before GroupConvolution unrolling Change-Id: Id94fb9c88a498d7b938f4f707e2e7b9b6df94880 --- src/graph/mutators/NodeFusionMutator.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src/graph/mutators/NodeFusionMutator.cpp') diff --git a/src/graph/mutators/NodeFusionMutator.cpp b/src/graph/mutators/NodeFusionMutator.cpp index 98c3a56018..9dc02d1ad1 100644 --- a/src/graph/mutators/NodeFusionMutator.cpp +++ b/src/graph/mutators/NodeFusionMutator.cpp @@ -73,13 +73,13 @@ void fuse_node_with_activation(Graph &g, ARM_COMPUTE_LOG_GRAPH_VERBOSE("Fusing node with ID : " << output_edge->producer_id() << " with Activation Layer node with ID : " << output_edge->consumer_id() << std::endl); - // Prevent fusion if batch normalization node has an output accessor + // Prevent fusion if fused node has an output accessor if(n_node->output(0)->accessor() == nullptr) { // Get driving nodes of activation node std::vector act_driving_nodes = get_driving_nodes(*act_node); - // Set activation info to batch normalization + // Set activation info to fused node n_node->set_fused_activation(act_node->activation_info()); // Extract activation node accessor if any @@ -88,13 +88,13 @@ void fuse_node_with_activation(Graph &g, // Remove activation node g.remove_node(act_node->id()); - // Update batch normalization node outputs + // Update fused node outputs for(auto &driving_node : act_driving_nodes) { g.add_connection(n_node->id(), 0, driving_node.node_id, driving_node.index); } - // Update accessor to batch normalization node + // Update accessor to fused node n_node->output(0)->set_accessor(std::move(act_node_accessor)); } else -- cgit v1.2.1