diff options
Diffstat (limited to 'src/backends/cl')
-rw-r--r-- | src/backends/cl/workloads/ClGatherWorkload.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/backends/cl/workloads/ClGatherWorkload.cpp b/src/backends/cl/workloads/ClGatherWorkload.cpp index b2341b8f32..06fa5af65e 100644 --- a/src/backends/cl/workloads/ClGatherWorkload.cpp +++ b/src/backends/cl/workloads/ClGatherWorkload.cpp @@ -37,7 +37,7 @@ ClGatherWorkload::ClGatherWorkload(const GatherQueueDescriptor& descriptor, info, this->GetGuid()); - m_Data.ValidateInputsOutputs("ClGatherWorkload", 1, 1); + m_Data.ValidateInputsOutputs("ClGatherWorkload", 2, 1); arm_compute::ICLTensor& input = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); arm_compute::ICLTensor& indices = static_cast<IClTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); |