diff options
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r-- | include/armnn/Descriptors.hpp | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 5d0990e816..653e64701a 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -711,6 +711,25 @@ struct FakeQuantizationDescriptor float m_Max; }; +/// A FillDescriptor for the FillLayer +struct FillDescriptor +{ + FillDescriptor() + : m_Value(0) + {} + + FillDescriptor(const float& value) + : m_Value(value) + {} + + bool operator ==(const FillDescriptor& rhs) const + { + return m_Value == rhs.m_Value; + } + + float m_Value; +}; + /// A ResizeBilinearDescriptor for the ResizeBilinearLayer. struct ResizeBilinearDescriptor { |