diff options
Diffstat (limited to 'src/backends/cl/workloads/ClRsqrtWorkload.cpp')
-rw-r--r-- | src/backends/cl/workloads/ClRsqrtWorkload.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/backends/cl/workloads/ClRsqrtWorkload.cpp b/src/backends/cl/workloads/ClRsqrtWorkload.cpp index 8d48bfad33..b8ae2f6d59 100644 --- a/src/backends/cl/workloads/ClRsqrtWorkload.cpp +++ b/src/backends/cl/workloads/ClRsqrtWorkload.cpp @@ -33,7 +33,10 @@ ClRsqrtWorkload::ClRsqrtWorkload(const RsqrtQueueDescriptor& descriptor, arm_compute::ICLTensor& input = PolymorphicDowncast<ClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); arm_compute::ICLTensor& output = PolymorphicDowncast<ClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor(); - m_RsqrtLayer.configure(clCompileContext, &input, &output); + { + ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "ClRsqrtWorkload_configure"); + m_RsqrtLayer.configure(clCompileContext, &input, &output); + } } void ClRsqrtWorkload::Execute() const |