diff options
Diffstat (limited to 'src/backends/backendsCommon/LayerSupportRules.hpp')
-rw-r--r-- | src/backends/backendsCommon/LayerSupportRules.hpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/LayerSupportRules.hpp b/src/backends/backendsCommon/LayerSupportRules.hpp index e616ecf022..a83fd62867 100644 --- a/src/backends/backendsCommon/LayerSupportRules.hpp +++ b/src/backends/backendsCommon/LayerSupportRules.hpp @@ -186,4 +186,12 @@ struct TensorNumDimensionsAreCorrect : public Rule } }; +struct TensorNumDimensionsAreGreaterOrEqualTo : public Rule +{ + TensorNumDimensionsAreGreaterOrEqualTo(const TensorInfo& info, unsigned int numDimensionsToCompare) + { + m_Res = info.GetNumDimensions() >= numDimensionsToCompare; + } +}; + } //namespace armnn
\ No newline at end of file |