diff options
author | Sheri Zhang <sheri.zhang@arm.com> | 2021-11-03 21:24:00 +0000 |
---|---|---|
committer | Sheri Zhang <sheri.zhang@arm.com> | 2021-11-04 15:35:58 +0000 |
commit | c65023e01d8e5c8f5af15dd8e2ca4fae3c4889f9 (patch) | |
tree | 313c5c0786971986d92c2e59073eb06406f43990 /arm_compute/graph/backends/FunctionHelpers.h | |
parent | cc859158faff318a81dd9aafcc5b609d7dd5d4a9 (diff) | |
download | ComputeLibrary-c65023e01d8e5c8f5af15dd8e2ca4fae3c4889f9.tar.gz |
Refine code for graph level fusion
Partially resolves: COMPMID-4701
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: I11fb89bd31ef5a98ad9a2d6bb5fd0c1e8371b52f
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6572
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/graph/backends/FunctionHelpers.h')
-rw-r--r-- | arm_compute/graph/backends/FunctionHelpers.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/arm_compute/graph/backends/FunctionHelpers.h b/arm_compute/graph/backends/FunctionHelpers.h index 6bec66a6ff..6aec3f6590 100644 --- a/arm_compute/graph/backends/FunctionHelpers.h +++ b/arm_compute/graph/backends/FunctionHelpers.h @@ -602,6 +602,7 @@ std::unique_ptr<IFunction> create_fused_convolution_with_post_op(FusedConvolutio std::unique_ptr<IFunction> func; std::string func_name; + // Fuse convolution with post ops is only supported for conv1x1, which is only implemented as gemmconv2d std::tie(func, func_name) = create_named_memory_managed_function<typename ConvolutionLayerFunctions::GEMMConvolutionLayer>( std::string("GEMMConvolutionLayer"), mm, input, weights, biases, output, conv_info, |