From 9b0a6b49e95b221456489dd7c58681ceca5dd8cb Mon Sep 17 00:00:00 2001 From: Viet-Hoa Do Date: Mon, 3 Apr 2023 16:27:25 +0100 Subject: Fix dynamic weights for CPU connected layer Resolves: COMPMID-5995 Signed-off-by: Viet-Hoa Do Change-Id: I707b8918bebee7e70d4de5207ef555c806e7a305 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9405 Benchmark: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: SiCong Li Reviewed-by: Jakub Sujak Comments-Addressed: Arm Jenkins --- src/cpu/operators/CpuGemmConv2d.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/cpu/operators/CpuGemmConv2d.cpp') diff --git a/src/cpu/operators/CpuGemmConv2d.cpp b/src/cpu/operators/CpuGemmConv2d.cpp index 9bf6ed1e85..ebf2ebcc1b 100644 --- a/src/cpu/operators/CpuGemmConv2d.cpp +++ b/src/cpu/operators/CpuGemmConv2d.cpp @@ -169,7 +169,7 @@ void CpuGemmConv2d::configure_mm(const ITensorInfo *src, const ITensorInfo *weig { // Configure matrix multiply function _mm_gemm = std::make_unique(); - _mm_gemm->configure(src, weights, biases, dst, 1.0f, 0.0f, gemm_info); + _mm_gemm->configure(src, weights, biases, dst, 1.0f, 1.0f, gemm_info); auto mm_mem_req = _mm_gemm->workspace(); for(unsigned int cont = 0; cont < mm_mem_req.size(); ++cont) { @@ -235,7 +235,7 @@ Status CpuGemmConv2d::validate_mm(const ITensorInfo *src, const ITensorInfo *wei else { // Perform validation step on Matrix multiply function - return CpuGemm::validate(src, weights, nullptr, dst, 1.0f, 0.0f, gemm_info); + return CpuGemm::validate(src, weights, biases, dst, 1.0f, 1.0f, gemm_info); } } -- cgit v1.2.1