ArmNN
 22.05.01
ShapesAreSameTotalSize Struct Reference

#include <LayerSupportRules.hpp>

Inheritance diagram for ShapesAreSameTotalSize:
Rule

Public Member Functions

 ShapesAreSameTotalSize (const TensorInfo &info0, const TensorInfo &info1)
 
- Public Member Functions inherited from Rule
bool operator() () const
 

Additional Inherited Members

- Public Attributes inherited from Rule
bool m_Res = true
 

Detailed Description

Definition at line 146 of file LayerSupportRules.hpp.

Constructor & Destructor Documentation

◆ ShapesAreSameTotalSize()

ShapesAreSameTotalSize ( const TensorInfo info0,
const TensorInfo info1 
)
inline

Definition at line 148 of file LayerSupportRules.hpp.

References TensorInfo::GetNumElements(), and Rule::m_Res.

149  {
150  m_Res = info0.GetNumElements() == info1.GetNumElements();
151  }

The documentation for this struct was generated from the following file: