mirror of https://github.com/Qortal/Brooklyn
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
1.8 KiB
63 lines
1.8 KiB
// |
|
// 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; |
|
}; |
|
|
|
|
|
|
|
|
|
|