diff options
Diffstat (limited to 'include')
-rw-r--r-- | include/armnn/Tensor.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/include/armnn/Tensor.hpp b/include/armnn/Tensor.hpp index 1bbc19f2f1..2326a0fc65 100644 --- a/include/armnn/Tensor.hpp +++ b/include/armnn/Tensor.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017,2022-2023 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017,2022-2024 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -188,6 +188,8 @@ public: bool operator==(const TensorInfo& other) const; bool operator!=(const TensorInfo& other) const; + using DifferenceType = std::vector<TensorInfo>::difference_type; + const TensorShape& GetShape() const { return m_Shape; } TensorShape& GetShape() { return m_Shape; } void SetShape(const TensorShape& newShape) { m_Shape = newShape; } |