aboutsummaryrefslogtreecommitdiff
path: root/src/cpu/operators/CpuGemmConv2d.cpp
diff options
context:
space:
mode:
authorViet-Hoa Do <viet-hoa.do@arm.com>2023-04-03 16:27:25 +0100
committerViet-Hoa Do <viet-hoa.do@arm.com>2023-04-14 08:57:27 +0000
commit9b0a6b49e95b221456489dd7c58681ceca5dd8cb (patch)
tree6afd87f8407fafb3de802e4ce1b4099a579b6ff8 /src/cpu/operators/CpuGemmConv2d.cpp
parent4e84f244548a18e0935502cc443336fc1b8f1454 (diff)
downloadComputeLibrary-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.cpp4
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);
}
}