aboutsummaryrefslogtreecommitdiff
path: root/test/TestTensor.cpp
blob: e6cb446fda30962828831fbb599d7d0eec8fbd89 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "TestTensor.hpp"

namespace driverTestHelpers
{

hidl_vec<uint32_t> TestTensor::GetDimensions() const
{
    hidl_vec<uint32_t> dimensions;
    dimensions.resize(m_Shape.GetNumDimensions());
    for (uint32_t i=0; i<m_Shape.GetNumDimensions(); ++i)
    {
        dimensions[i] = m_Shape[i];
    }
    return dimensions;
}

unsigned int TestTensor::GetNumElements() const
{
    return m_Shape.GetNumElements();
}

const float * TestTensor::GetData() const
{
    ARMNN_ASSERT(m_Data.empty() == false);
    return &m_Data[0];
}

} // namespace driverTestHelpers