diff options
Diffstat (limited to 'src/backends/ClWorkloads/ClConvolution2dUint8Workload.cpp')
| -rw-r--r-- | src/backends/ClWorkloads/ClConvolution2dUint8Workload.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/ClWorkloads/ClConvolution2dUint8Workload.cpp b/src/backends/ClWorkloads/ClConvolution2dUint8Workload.cpp index e6783b698a..c9f5eaa31d 100644 --- a/src/backends/ClWorkloads/ClConvolution2dUint8Workload.cpp +++ b/src/backends/ClWorkloads/ClConvolution2dUint8Workload.cpp @@ -51,11 +51,11 @@ ClConvolution2dUint8Workload::ClConvolution2dUint8Workload(const Convolution2dQu &output, padStrideInfo); - InitialiseArmComputeClTensorData(*m_KernelTensor, m_Data.m_Weight->GetConstTensor<uint8_t>()); + InitializeArmComputeClTensorData(*m_KernelTensor, m_Data.m_Weight); if (m_BiasTensor) { - InitialiseArmComputeClTensorData(*m_BiasTensor, m_Data.m_Bias->GetConstTensor<int32_t>()); + InitializeArmComputeClTensorData(*m_BiasTensor, m_Data.m_Bias); } // Force Compute Library to perform the necessary copying and reshaping, after which |