diff options
author | Jim Flynn <jim.flynn@arm.com> | 2022-02-14 11:22:29 +0000 |
---|---|---|
committer | Colm Donelan <colm.donelan@arm.com> | 2022-02-16 21:36:34 +0000 |
commit | a00bad1140223453e72a134388b209f9fa296d00 (patch) | |
tree | 81774e9748b0726981fc3160d1fd232bc6359512 /src/backends/cl/workloads/ClConvertFp16ToFp32Workload.cpp | |
parent | 01055eba0287778f509ac292aa6fa068980e2034 (diff) | |
download | armnn-a00bad1140223453e72a134388b209f9fa296d00.tar.gz |
Add implementation of reconfigure function to Cl Convert workloads
Change-Id: Id993a300847bde012dd38ef7f44fcd3c14867349
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Diffstat (limited to 'src/backends/cl/workloads/ClConvertFp16ToFp32Workload.cpp')
-rw-r--r-- | src/backends/cl/workloads/ClConvertFp16ToFp32Workload.cpp | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/src/backends/cl/workloads/ClConvertFp16ToFp32Workload.cpp b/src/backends/cl/workloads/ClConvertFp16ToFp32Workload.cpp index d4110df0ae..8ccf157aca 100644 --- a/src/backends/cl/workloads/ClConvertFp16ToFp32Workload.cpp +++ b/src/backends/cl/workloads/ClConvertFp16ToFp32Workload.cpp @@ -25,9 +25,13 @@ ClConvertFp16ToFp32Workload::ClConvertFp16ToFp32Workload( arm_compute::ICLTensor& input = static_cast<IClTensorHandle*>(this->m_Data.m_Inputs[0])->GetTensor(); arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(this->m_Data.m_Outputs[0])->GetTensor(); + // Create Proxy tensor and set the initial tensor handle to it + m_InputProxy = std::make_unique<ICLTensorProxy>(&input); + m_OutputProxy = std::make_unique<ICLTensorProxy>(&output); + { ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "ClConvertFp16ToFp32Workload_configure"); - m_Layer.configure(clCompileContext, &input, &output, g_AclConvertPolicy, 0); + m_Layer.configure(clCompileContext, m_InputProxy.get(), m_OutputProxy.get(), g_AclConvertPolicy, 0); } } @@ -92,7 +96,10 @@ void ClConvertFp16ToFp32Workload::ReplaceOutputTensorHandle(ITensorHandle* tenso void ClConvertFp16ToFp32Workload::Reconfigure() { - throw armnn::UnimplementedException("Reconfigure not implemented for this workload"); + arm_compute::ICLTensor& input = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); + arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); + m_InputProxy->set(&input); + m_OutputProxy->set(&output); } } //namespace armnn |