diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-11-02 16:43:57 +0000 |
---|---|---|
committer | Colm Donelan <colm.donelan@arm.com> | 2021-11-03 14:03:03 +0000 |
commit | 2ea38c7f56d57a4fdcf709c9d61b7bdfab4ebfd9 (patch) | |
tree | d422855474c36758a4616c8ce3b7ad33510acef5 | |
parent | f39f8d8597c59057118f67cacf70d246f95fea9b (diff) | |
download | armnn-2ea38c7f56d57a4fdcf709c9d61b7bdfab4ebfd9.tar.gz |
Bug Fix in CpuAcc and GpuAcc workload: Gather has 2 inputs
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ifbb9332c78c843c5b937c4e1b50a8f5a75409e73
-rw-r--r-- | src/backends/cl/workloads/ClGatherWorkload.cpp | 2 | ||||
-rw-r--r-- | src/backends/neon/workloads/NeonGatherWorkload.cpp | 2 |
2 files changed, 2 insertions, 2 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(); diff --git a/src/backends/neon/workloads/NeonGatherWorkload.cpp b/src/backends/neon/workloads/NeonGatherWorkload.cpp index f5c8d34235..4cb17caffa 100644 --- a/src/backends/neon/workloads/NeonGatherWorkload.cpp +++ b/src/backends/neon/workloads/NeonGatherWorkload.cpp @@ -34,7 +34,7 @@ NeonGatherWorkload::NeonGatherWorkload(const GatherQueueDescriptor& descriptor, info, this->GetGuid()); - m_Data.ValidateInputsOutputs("NeonGatherWorkload", 1, 1); + m_Data.ValidateInputsOutputs("NeonGatherWorkload", 2, 1); arm_compute::ITensor& input = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor(); arm_compute::ITensor& indices = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTensor(); |