diff options
Diffstat (limited to 'test/TestTensor.cpp')
-rw-r--r-- | test/TestTensor.cpp | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/test/TestTensor.cpp b/test/TestTensor.cpp new file mode 100644 index 00000000..0766ef50 --- /dev/null +++ b/test/TestTensor.cpp @@ -0,0 +1,32 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// See LICENSE file in the project root for full license information. +// +#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 +{ + BOOST_ASSERT(m_Data.empty() == false); + return &m_Data[0]; +} + +} // namespace driverTestHelpers |