diff options
author | Kevin May <kevin.may@arm.com> | 2021-10-22 15:42:28 +0100 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2021-11-01 09:46:40 +0000 |
commit | 9f6862de94e3d15ea5207a5747012f6c7eead358 (patch) | |
tree | 4b45cab325cd7eb36fd04a1cba7d0e7a3ec92dbb /src/backends/cl/workloads/ClQuantizedLstmWorkload.cpp | |
parent | b1c62f11881e0d528bea5b3664a8f36e4c03b508 (diff) | |
download | armnn-9f6862de94e3d15ea5207a5747012f6c7eead358.tar.gz |
IVGCVSW-6440 Add profiling around CL workload configure calls
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: I7626d5bd82e832d5be6913719a34d76fbd1dbed8
Diffstat (limited to 'src/backends/cl/workloads/ClQuantizedLstmWorkload.cpp')
-rw-r--r-- | src/backends/cl/workloads/ClQuantizedLstmWorkload.cpp | 20 |
1 files changed, 12 insertions, 8 deletions
diff --git a/src/backends/cl/workloads/ClQuantizedLstmWorkload.cpp b/src/backends/cl/workloads/ClQuantizedLstmWorkload.cpp index 7bacf70a6a..05ae89c93b 100644 --- a/src/backends/cl/workloads/ClQuantizedLstmWorkload.cpp +++ b/src/backends/cl/workloads/ClQuantizedLstmWorkload.cpp @@ -109,14 +109,18 @@ ClQuantizedLstmWorkload::ClQuantizedLstmWorkload(const QuantizedLstmQueueDescrip arm_compute::ICLTensor& cellStateOutTensor = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); arm_compute::ICLTensor& outputStateOutTensor = static_cast<IClTensorHandle*>(m_Data.m_Outputs[1])->GetTensor(); - m_QuantizedLstmLayer.configure(clCompileContext, &inputTensor, m_InputToInputWeightsTensor.get(), - m_InputToForgetWeightsTensor.get(), - m_InputToCellWeightsTensor.get(), m_InputToOutputWeightsTensor.get(), - m_RecurrentToInputWeightsTensor.get(), m_RecurrentToForgetWeightsTensor.get(), - m_RecurrentToCellWeightsTensor.get(), m_RecurrentToOutputWeightsTensor.get(), - m_InputGateBiasTensor.get(), m_ForgetGateBiasTensor.get(), m_CellBiasTensor.get(), - m_OutputGateBiasTensor.get(), &cellStateInTensor, &outputStateInTensor, - &cellStateOutTensor, &outputStateOutTensor); + { + ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "ClQuantizedLstmWorkload_configure"); + m_QuantizedLstmLayer.configure(clCompileContext, &inputTensor, m_InputToInputWeightsTensor.get(), + m_InputToForgetWeightsTensor.get(), + m_InputToCellWeightsTensor.get(), m_InputToOutputWeightsTensor.get(), + m_RecurrentToInputWeightsTensor.get(), m_RecurrentToForgetWeightsTensor.get(), + m_RecurrentToCellWeightsTensor.get(), m_RecurrentToOutputWeightsTensor.get(), + m_InputGateBiasTensor.get(), m_ForgetGateBiasTensor.get(), + m_CellBiasTensor.get(), + m_OutputGateBiasTensor.get(), &cellStateInTensor, &outputStateInTensor, + &cellStateOutTensor, &outputStateOutTensor); + } InitializeArmComputeClTensorData(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights); InitializeArmComputeClTensorData(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights); |