LayerTestResult.hpp File Reference

#include <armnn/Tensor.hpp>
#include <armnn/utility/Assert.hpp>
#include <boost/multi_array.hpp>
#include <cstddef>

Go to the source code of this file.

Classes
struct	LayerTestResult< T, n >

Functions
template<std::size_t n>
boost::array< unsigned int, n >	GetTensorShapeAsArray (const armnn::TensorInfo &tensorInfo)

Function Documentation

GetTensorShapeAsArray()

boost::array<unsigned int, n> GetTensorShapeAsArray

(

const armnn::TensorInfo &

tensorInfo

)

Definition at line 16 of file LayerTestResult.hpp.

References ARMNN_ASSERT_MSG, TensorInfo::GetNumDimensions(), and TensorInfo::GetShape().

{
    ARMNN_ASSERT_MSG(n == tensorInfo.GetNumDimensions(),
        "Attempting to construct a shape array of mismatching size");

    boost::array<unsigned int, n> shape;
    for (unsigned int i = 0; i < n; i++)
    {
        shape[i] = tensorInfo.GetShape()[i];
    }
    return shape;
}