1 files changed, 40 insertions, 0 deletions
diff --git a/src/backends/tosaCommon/TosaLayerSupportRules.hpp b/src/backends/tosaCommon/TosaLayerSupportRules.hpp
new file mode 100644
index 0000000000..2a2b08da99
--- /dev/null
+++ b/src/backends/tosaCommon/TosaLayerSupportRules.hpp
@@ -0,0 +1,40 @@
+//
+// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+// List of Layer Support Rules common to TOSA backends only, for use with CheckSupportRule()
+
+struct TosaOperatorAttributeOfAny : public Rule
+{
+    template<typename Container>
+    explicit TosaOperatorAttributeOfAny(TosaSerializationOperator* op, const Container& c)
+    {
+        m_Res = std::any_of(c.begin(), c.end(), [&op](Attribute attribute)
+        {
+            return attribute == op->GetAttributeType();
+        });
+    }
+};
+
+struct TosaTypeAnyOf : public Rule
+{
+    template<typename Container>
+    TosaTypeAnyOf(TosaSerializationTensor* tensor, const Container& c)
+    {
+        m_Res = std::any_of(c.begin(), c.end(), [&tensor](DType dt)
+        {
+            return dt == tensor->GetDtype();
+        });
+    }
+};
+
+struct TosaTensorNumDimensionsWithinBounds : public Rule
+{
+    explicit TosaTensorNumDimensionsWithinBounds(TosaSerializationTensor* tensor)
+    {
+        m_Res = (tensor->GetShape().size() <= MaxNumOfTensorDimensions) || (!tensor->GetShape().empty());
+    }
+};