diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-03-29 10:54:36 +0000 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-03-29 14:00:24 +0000 |
commit | cadb368b0827601647c3d1fd66689f96473af5cb (patch) | |
tree | bdbb1f792af7dfdea2644ef01a4538e178f824c4 /src/graph/mutators/NodeFusionMutator.cpp | |
parent | 0bc784982f183d9d50be31adb867e84c237d9fc3 (diff) | |
download | ComputeLibrary-cadb368b0827601647c3d1fd66689f96473af5cb.tar.gz |
COMPMID-1995: Fixed graph fusion mutator for float types.
-Fixes precondition checks for fusing activation with other nodes.
-Fixes is_relu6 check to capture bounded relu as well.
Change-Id: Iba193af51491b537c884a35ca85172151534f3ec
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/918
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/graph/mutators/NodeFusionMutator.cpp')
-rw-r--r-- | src/graph/mutators/NodeFusionMutator.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/graph/mutators/NodeFusionMutator.cpp b/src/graph/mutators/NodeFusionMutator.cpp index 724307e7b7..b28f2dbd2e 100644 --- a/src/graph/mutators/NodeFusionMutator.cpp +++ b/src/graph/mutators/NodeFusionMutator.cpp @@ -221,7 +221,7 @@ void NodeFusionMutator::mutate(Graph &g) const bool same_qinfo = n.output(0)->desc().quant_info == output_edge->producer()->output(0)->desc().quant_info; const bool output_qasymm8 = n.output(0)->desc().data_type == DataType::QASYMM8; - return output_qasymm8 && same_qinfo; + return (output_qasymm8 && same_qinfo) || !output_qasymm8; }; // Fusion mutations |