diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-11-12 18:36:19 +0000 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-11-14 16:00:45 +0000 |
commit | 1c32bf396eb690a54fd94487e3f258b2c7d31753 (patch) | |
tree | 83dafe008fe428133c1c531cd179e4cad256ef5c /src/graph/mutators/NodeFusionMutator.cpp | |
parent | 6c7c38e70c795077ba727aadeefc670888bec089 (diff) | |
download | ComputeLibrary-1c32bf396eb690a54fd94487e3f258b2c7d31753.tar.gz |
COMPMID-1451: Perform fusion before GroupConvolution unrolling
Change-Id: Id94fb9c88a498d7b938f4f707e2e7b9b6df94880
Diffstat (limited to 'src/graph/mutators/NodeFusionMutator.cpp')
-rw-r--r-- | src/graph/mutators/NodeFusionMutator.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
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<NodeIdxPair> 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 |