diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-04-18 12:48:46 +0100 |
---|---|---|
committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-05-08 18:18:46 +0100 |
commit | 6fba29488f6053cb28be4d7e7a99cf5ecd41c08f (patch) | |
tree | f6ac9ef2b5de6f5b2c0b90fdf8e28b3a2d0f71c0 /src/backends/neon/workloads/NeonConvolution2dWorkload.cpp | |
parent | ee1497c400db10134ab540005e105f64bd0f486a (diff) | |
download | armnn-6fba29488f6053cb28be4d7e7a99cf5ecd41c08f.tar.gz |
IVGCVSW-7454 Fix CpuAcc FC dynamic weights
* Pass to ACL the flag for constant weights and bias in FC, conv and DWconv workloads
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Iae2810c8d1a402d4afc1e757846665315a80d3ea
Diffstat (limited to 'src/backends/neon/workloads/NeonConvolution2dWorkload.cpp')
-rw-r--r-- | src/backends/neon/workloads/NeonConvolution2dWorkload.cpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/backends/neon/workloads/NeonConvolution2dWorkload.cpp b/src/backends/neon/workloads/NeonConvolution2dWorkload.cpp index 879bb747a1..3f69718e8f 100644 --- a/src/backends/neon/workloads/NeonConvolution2dWorkload.cpp +++ b/src/backends/neon/workloads/NeonConvolution2dWorkload.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2017,2022 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017,2022-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -91,10 +91,14 @@ NeonConvolution2dWorkload::NeonConvolution2dWorkload( m_KernelTensor = std::make_unique<arm_compute::Tensor>(); BuildArmComputeTensor(*m_KernelTensor, info.m_InputTensorInfos[1], m_Data.m_Parameters.m_DataLayout); + m_KernelTensor->info()->set_are_values_constant(info.m_InputTensorInfos[1].IsConstant()); if (m_Data.m_Parameters.m_BiasEnabled) { m_BiasTensor = std::make_unique<arm_compute::Tensor>(); BuildArmComputeTensor(*m_BiasTensor, info.m_InputTensorInfos[2], m_Data.m_Parameters.m_DataLayout); + m_BiasTensor->info()->set_are_values_constant(info.m_InputTensorInfos[2].IsConstant()); + // We do not support dynamic bias + ARMNN_ASSERT(info.m_InputTensorInfos[2].IsConstant() == true); } arm_compute::PadStrideInfo padStrideInfo = BuildArmComputePadStrideInfo(m_Data.m_Parameters); |