diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-05-09 16:25:47 +0100 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-05-09 18:33:51 +0100 |
| commit | b190da2779b0419472c0734176c703a252d47b8b (patch) | |
| tree | 6a42afb11551c08bdee80f2e22676d5b87bd6805 /src/backends/neon/workloads/NeonGatherNdWorkload.hpp | |
| parent | 84d63785eb2dceba297a685ebd98f1d29be47326 (diff) | |
| download | armnn-b190da2779b0419472c0734176c703a252d47b8b.tar.gz | |
IVGCVSW-6862 Use same datatype for all containers of indices in NeonGatherNd
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I6b1c7c1c499dc93aa58fa9f58b64fb664e8bcc56
Diffstat (limited to 'src/backends/neon/workloads/NeonGatherNdWorkload.hpp')
| -rw-r--r-- | src/backends/neon/workloads/NeonGatherNdWorkload.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/neon/workloads/NeonGatherNdWorkload.hpp b/src/backends/neon/workloads/NeonGatherNdWorkload.hpp index 848aac667b..e9588f4d41 100644 --- a/src/backends/neon/workloads/NeonGatherNdWorkload.hpp +++ b/src/backends/neon/workloads/NeonGatherNdWorkload.hpp @@ -27,9 +27,9 @@ public: private: arm_compute::Tensor m_FlattenedCoeff; - arm_compute::Tensor m_outputMul; + arm_compute::Tensor m_OutputMul; arm_compute::Tensor m_FlattenedIndices; - arm_compute::Tensor m_outputGather; + arm_compute::Tensor m_OutputGather; mutable arm_compute::NEPixelWiseMultiplication m_MulLayer; mutable arm_compute::NEReductionOperation m_ReduceSumLayer; |