mirror of
https://github.com/Qortal/Brooklyn.git
synced 2025-02-21 22:55:53 +00:00
64 lines
1.8 KiB
C++
64 lines
1.8 KiB
C++
//
|
|
// 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;
|
|
};
|
|
|
|
|
|
|
|
|