diff options
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/Tensor.cpp | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/src/armnn/Tensor.cpp b/src/armnn/Tensor.cpp index 8e72d4694c..6e09e3bc59 100644 --- a/src/armnn/Tensor.cpp +++ b/src/armnn/Tensor.cpp @@ -23,6 +23,22 @@ TensorShape::TensorShape() { } +TensorShape::TensorShape(unsigned int numDimensions) + : m_NumDimensions(numDimensions) +{ + if (numDimensions < 1) + { + throw InvalidArgumentException("Tensor numDimensions must be greater than 0"); + } + + if (numDimensions > MaxNumOfTensorDimensions) + { + throw InvalidArgumentException("Tensor numDimensions must be less than or equal to MaxNumOfTensorDimensions"); + } + + std::fill(m_Dimensions.begin(), m_Dimensions.begin() + m_NumDimensions, 0); +} + TensorShape::TensorShape(const unsigned int numDimensions, const unsigned int* const dimensionSizes) : m_NumDimensions(numDimensions) { |