aboutsummaryrefslogtreecommitdiff
path: root/tests/LstmCommon.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/LstmCommon.hpp')
-rwxr-xr-xtests/LstmCommon.hpp30
1 files changed, 30 insertions, 0 deletions
diff --git a/tests/LstmCommon.hpp b/tests/LstmCommon.hpp
new file mode 100755
index 0000000000..31c4d041c1
--- /dev/null
+++ b/tests/LstmCommon.hpp
@@ -0,0 +1,30 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+#pragma once
+
+#include <string>
+#include <utility>
+
+namespace
+{
+
+struct LstmInput
+{
+ LstmInput(const std::vector<float>& inputSeq,
+ const std::vector<float>& stateC,
+ const std::vector<float>& stateH)
+ : m_InputSeq(inputSeq)
+ , m_StateC(stateC)
+ , m_StateH(stateH)
+ {}
+
+ std::vector<float> m_InputSeq;
+ std::vector<float> m_StateC;
+ std::vector<float> m_StateH;
+};
+
+using LstmInputs = std::pair<std::string, std::vector<LstmInput>>;
+
+} // anonymous namespace \ No newline at end of file