From b37e1facda8f0b65a77f03a0cb2cbac4fab941ba Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Thu, 19 Oct 2023 13:41:50 +0000
Subject: [PATCH] [Add] Attributes names for Concat Operator
---
include/aidge/operator/Concat.hpp | 10 +++++++++-
1 file changed, 9 insertions(+), 1 deletion(-)
diff --git a/include/aidge/operator/Concat.hpp b/include/aidge/operator/Concat.hpp
index 20ae1be63..6943b3594 100644
--- a/include/aidge/operator/Concat.hpp
+++ b/include/aidge/operator/Concat.hpp
@@ -26,7 +26,7 @@
#include "aidge/utils/Types.h"
namespace Aidge {
-enum class ConcatAttr { NbInputs, Axis};
+enum class ConcatAttr { NbInputs, Axis };
class Concat_Op : public Operator,
public Registrable<Concat_Op, std::string, std::unique_ptr<OperatorImpl>(const Concat_Op&)>,
@@ -183,4 +183,12 @@ inline std::shared_ptr<Node> Concat(const IOIndex_t nbIn, const DimIdx_t axis =
}
}
+namespace {
+ template <>
+ const char* const EnumStrings<Aidge::ConcatAttr>::data[] = {
+ "NbInputs",
+ "Axis"
+ };
+}
+
#endif /* AIDGE_CORE_OPERATOR_CONCAT_H_ */
--
GitLab