xref: /aosp_15_r20/external/android-nn-driver/test/TestHalfTensor.cpp (revision 3e777be0405cee09af5d5785ff37f7cfb5bee59a) 
1*3e777be0SXin Li //
2*3e777be0SXin Li // Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
3*3e777be0SXin Li // SPDX-License-Identifier: MIT
4*3e777be0SXin Li //
5*3e777be0SXin Li 
6*3e777be0SXin Li #include "TestHalfTensor.hpp"
7*3e777be0SXin Li 
8*3e777be0SXin Li namespace driverTestHelpers
9*3e777be0SXin Li {
10*3e777be0SXin Li 
GetDimensions() const11*3e777be0SXin Li hidl_vec<uint32_t> TestHalfTensor::GetDimensions() const
12*3e777be0SXin Li {
13*3e777be0SXin Li     hidl_vec<uint32_t> dimensions;
14*3e777be0SXin Li     dimensions.resize(m_Shape.GetNumDimensions());
15*3e777be0SXin Li     for (uint32_t i=0; i<m_Shape.GetNumDimensions(); ++i)
16*3e777be0SXin Li     {
17*3e777be0SXin Li         dimensions[i] = m_Shape[i];
18*3e777be0SXin Li     }
19*3e777be0SXin Li     return dimensions;
20*3e777be0SXin Li }
21*3e777be0SXin Li 
GetNumElements() const22*3e777be0SXin Li unsigned int TestHalfTensor::GetNumElements() const
23*3e777be0SXin Li {
24*3e777be0SXin Li     return m_Shape.GetNumElements();
25*3e777be0SXin Li }
26*3e777be0SXin Li 
GetData() const27*3e777be0SXin Li const Half * TestHalfTensor::GetData() const
28*3e777be0SXin Li {
29*3e777be0SXin Li     DOCTEST_CHECK(m_Data.empty() == false);
30*3e777be0SXin Li     return &m_Data[0];
31*3e777be0SXin Li }
32*3e777be0SXin Li 
33*3e777be0SXin Li } // namespace driverTestHelpers
34