diff options
author | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-04-03 16:27:25 +0100 |
---|---|---|
committer | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-04-14 08:57:27 +0000 |
commit | 9b0a6b49e95b221456489dd7c58681ceca5dd8cb (patch) | |
tree | 6afd87f8407fafb3de802e4ce1b4099a579b6ff8 /src/cpu/operators/CpuGemmConv2d.cpp | |
parent | 4e84f244548a18e0935502cc443336fc1b8f1454 (diff) | |
download | ComputeLibrary-9b0a6b49e95b221456489dd7c58681ceca5dd8cb.tar.gz |
Fix dynamic weights for CPU connected layer
Resolves: COMPMID-5995
Signed-off-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Change-Id: I707b8918bebee7e70d4de5207ef555c806e7a305
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9405
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Reviewed-by: Jakub Sujak <jakub.sujak@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/operators/CpuGemmConv2d.cpp')
-rw-r--r-- | src/cpu/operators/CpuGemmConv2d.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
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<CpuGemm>(); - _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); } } |