Skip to content
Snippets Groups Projects
Commit b37e1fac authored by Maxence Naud's avatar Maxence Naud
Browse files

[Add] Attributes names for Concat Operator

parent fe1b11fa
No related branches found
No related tags found
No related merge requests found
......@@ -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_ */
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment