35 lines
670 B
C++
35 lines
670 B
C++
//
|
|
// Copyright © 2017 Arm Ltd. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
//
|
|
#pragma once
|
|
|
|
#include "../ArmnnDriver.hpp"
|
|
|
|
#include <armnn/utility/Assert.hpp>
|
|
|
|
namespace driverTestHelpers
|
|
{
|
|
|
|
class TestTensor
|
|
{
|
|
public:
|
|
TestTensor(const armnn::TensorShape & shape,
|
|
const std::vector<float> & data)
|
|
: m_Shape{shape}
|
|
, m_Data{data}
|
|
{
|
|
ARMNN_ASSERT(m_Shape.GetNumElements() == m_Data.size());
|
|
}
|
|
|
|
hidl_vec<uint32_t> GetDimensions() const;
|
|
unsigned int GetNumElements() const;
|
|
const float * GetData() const;
|
|
|
|
private:
|
|
armnn::TensorShape m_Shape;
|
|
std::vector<float> m_Data;
|
|
};
|
|
|
|
} // driverTestHelpers
|