diff options
Diffstat (limited to 'src/armnn/Descriptors.cpp')
| -rw-r--r-- | src/armnn/Descriptors.cpp | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/src/armnn/Descriptors.cpp b/src/armnn/Descriptors.cpp index a1419cfbf7..e6374aea8f 100644 --- a/src/armnn/Descriptors.cpp +++ b/src/armnn/Descriptors.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #include "armnn/Descriptors.hpp" @@ -363,6 +363,24 @@ void swap(ViewsDescriptor& first, ViewsDescriptor& second) swap(first.m_ViewSizes, second.m_ViewSizes); } +void ViewsDescriptor::SetAxis(int32_t axis) +{ + m_Axis = axis; + m_IsAxisSet = true; +} + +/// Get the axis value. +int32_t ViewsDescriptor::GetAxis() const +{ + return m_Axis; +} + +/// Returns true if an axis has been set. +bool ViewsDescriptor::HasAxis() const +{ + return m_IsAxisSet; +} + int StridedSliceDescriptor::GetStartForAxis(const TensorShape& inputShape, unsigned int axis) const { |