//
// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//

#include "TestHalfTensor.hpp"

namespace driverTestHelpers
{

hidl_vec<uint32_t> TestHalfTensor::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 TestHalfTensor::GetNumElements() const
{
    return m_Shape.GetNumElements();
}

const Half * TestHalfTensor::GetData() const
{
    DOCTEST_CHECK(m_Data.empty() == false);
    return &m_Data[0];
}

} // namespace driverTestHelpers