blob: a8045c5b15c4113e392c509c23e240ce090dfcc6 [file] [log] [blame]
surmeh0149b9e102018-05-17 14:11:25 +01001//
2// Copyright © 2017 Arm Ltd. All rights reserved.
David Beck93e48982018-09-05 13:05:09 +01003// SPDX-License-Identifier: MIT
surmeh0149b9e102018-05-17 14:11:25 +01004//
5#include "TestTensor.hpp"
6
7namespace driverTestHelpers
8{
9
10hidl_vec<uint32_t> TestTensor::GetDimensions() const
11{
12 hidl_vec<uint32_t> dimensions;
13 dimensions.resize(m_Shape.GetNumDimensions());
14 for (uint32_t i=0; i<m_Shape.GetNumDimensions(); ++i)
15 {
16 dimensions[i] = m_Shape[i];
17 }
18 return dimensions;
19}
20
21unsigned int TestTensor::GetNumElements() const
22{
23 return m_Shape.GetNumElements();
24}
25
26const float * TestTensor::GetData() const
27{
28 BOOST_ASSERT(m_Data.empty() == false);
29 return &m_Data[0];
30}
31
32} // namespace driverTestHelpers