diff options
Diffstat (limited to 'src/armnn/Descriptors.cpp')
-rw-r--r-- | src/armnn/Descriptors.cpp | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/src/armnn/Descriptors.cpp b/src/armnn/Descriptors.cpp index a6339cf08e..381d040683 100644 --- a/src/armnn/Descriptors.cpp +++ b/src/armnn/Descriptors.cpp @@ -124,6 +124,29 @@ OriginsDescriptor& OriginsDescriptor::operator=(OriginsDescriptor rhs) return *this; } +bool OriginsDescriptor::operator==(const OriginsDescriptor& rhs) const +{ + if (GetNumViews() != rhs.GetNumViews() || + GetNumDimensions() != rhs.GetNumDimensions() || + GetConcatAxis() != rhs.GetConcatAxis()) + { + return false; + } + + for (unsigned int i = 0u; i < GetNumViews(); ++i) + { + for (unsigned int j = 0u; j < GetNumDimensions(); ++j) + { + if (GetViewOrigin(i)[j] != rhs.GetViewOrigin(i)[j]) + { + return false; + } + } + } + + return true; +} + void OriginsDescriptor::SetConcatAxis(unsigned int concatAxis) { m_ConcatAxis = concatAxis; @@ -240,6 +263,27 @@ ViewsDescriptor& ViewsDescriptor::operator=(ViewsDescriptor rhs) return *this; } +bool ViewsDescriptor::operator==(const ViewsDescriptor& rhs) const +{ + if (GetNumViews() != rhs.GetNumViews() || GetNumDimensions() != rhs.GetNumDimensions()) + { + return false; + } + + for (unsigned int i = 0u; i < GetNumViews(); ++i) + { + for (unsigned int j = 0u; j < GetNumDimensions(); ++j) + { + if (GetViewOrigin(i)[j] != rhs.GetViewOrigin(i)[j] || GetViewSizes(i)[j] != rhs.GetViewSizes(i)[j]) + { + return false; + } + } + } + + return true; +} + uint32_t ViewsDescriptor::GetNumViews() const { return m_Origins.GetNumViews(); |