//
// 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