1 files changed, 63 insertions, 0 deletions
diff --git a/include/armnnTestUtils/LayerTestResult.hpp b/include/armnnTestUtils/LayerTestResult.hpp
new file mode 100644
index 0000000000..410973e4b1
--- /dev/null
+++ b/include/armnnTestUtils/LayerTestResult.hpp
@@ -0,0 +1,63 @@
+//
+// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include <armnn/Tensor.hpp>
+#include <armnn/utility/Assert.hpp>
+
+#include <cstddef>
+#include <vector>
+
+template <typename T, std::size_t n>
+struct LayerTestResult
+{
+    LayerTestResult(const armnn::TensorInfo& outputInfo)
+        : m_Supported(true)
+        , m_CompareBoolean(false)
+    {
+        m_ActualData.reserve(outputInfo.GetNumElements());
+        m_ExpectedData.reserve(outputInfo.GetNumElements());
+        m_ActualShape = outputInfo.GetShape();
+        m_ExpectedShape = outputInfo.GetShape();
+    }
+
+    LayerTestResult(const std::vector<T>& actualData,
+                    const std::vector<T>& expectedData,
+                    const armnn::TensorShape& actualShape,
+                    const armnn::TensorShape& expectedShape)
+        : m_ActualData(actualData)
+        , m_ExpectedData(expectedData)
+        , m_ActualShape(actualShape)
+        , m_ExpectedShape(expectedShape)
+        , m_Supported(true)
+        , m_CompareBoolean(false)
+    {}
+
+    LayerTestResult(const std::vector<T>& actualData,
+                    const std::vector<T>& expectedData,
+                    const armnn::TensorShape& actualShape,
+                    const armnn::TensorShape& expectedShape,
+                    const bool compareBoolean)
+        : m_ActualData(actualData)
+        , m_ExpectedData(expectedData)
+        , m_ActualShape(actualShape)
+        , m_ExpectedShape(expectedShape)
+        , m_Supported(true)
+        , m_CompareBoolean(compareBoolean)
+    {}
+
+    std::vector<T> m_ActualData;
+    std::vector<T> m_ExpectedData;
+    armnn::TensorShape m_ActualShape;
+    armnn::TensorShape m_ExpectedShape;
+
+    bool m_Supported;
+    bool m_CompareBoolean;
+};
+
+
+
+