From ad9171701e6032b3ddf3573f85780bae30c512c6 Mon Sep 17 00:00:00 2001
From: Cathal Corbett <cathal.corbett@arm.com>
Date: Wed, 9 Feb 2022 23:21:35 +0000
Subject: IVGCVSW-6267 Add support of Unidirectional Sequence Lstm fp32/fp16 to
 Cl

!ComputeLibrary:7150

Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: I01690e6555978d93c41d09bbe5378683bc925f61
---
 .../neon/workloads/NeonUnidirectionalSequenceLstmFloatWorkload.hpp     | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'src/backends/neon/workloads/NeonUnidirectionalSequenceLstmFloatWorkload.hpp')

diff --git a/src/backends/neon/workloads/NeonUnidirectionalSequenceLstmFloatWorkload.hpp b/src/backends/neon/workloads/NeonUnidirectionalSequenceLstmFloatWorkload.hpp
index 10c2ecbd19..776afd3965 100644
--- a/src/backends/neon/workloads/NeonUnidirectionalSequenceLstmFloatWorkload.hpp
+++ b/src/backends/neon/workloads/NeonUnidirectionalSequenceLstmFloatWorkload.hpp
@@ -31,10 +31,13 @@ private:
     //
     // ACL layers required to fully form a Unidirectional Sequence LSTM layer.
     //
+
+    // permutation for input (only used when input is batch major)
     mutable std::unique_ptr<arm_compute::NEPermute> m_Permute1;
     mutable std::unique_ptr<arm_compute::IFunction> m_Splitter;
     mutable std::vector<std::unique_ptr<arm_compute::NELSTMLayer>> m_Layers;
     mutable std::unique_ptr<arm_compute::NEConcatenateLayer> m_Concat;
+    // permutation for output (only used when input is batch major)
     mutable std::unique_ptr<arm_compute::NEPermute> m_Permute2;
 
     //
-- 
cgit v1.2.1