diff options
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r-- | include/armnn/Descriptors.hpp | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 653e64701a..60aa219638 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -730,6 +730,26 @@ struct FillDescriptor float m_Value; }; +/// A GatherDescriptor for the GatherLayer. +struct GatherDescriptor +{ + GatherDescriptor() + : m_Axis(0) + {} + + GatherDescriptor(int32_t axis) + : m_Axis(axis) + {} + + bool operator ==(const GatherDescriptor& rhs) const + { + return m_Axis == rhs.m_Axis; + } + + /// The axis in params to gather indices from + int32_t m_Axis; +}; + /// A ResizeBilinearDescriptor for the ResizeBilinearLayer. struct ResizeBilinearDescriptor { |