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
2 merge requests!46Remove Operator reference to Tensor,!20Draft: Introduction of Tiling
Pipeline #33072 canceled
...@@ -26,7 +26,7 @@ ...@@ -26,7 +26,7 @@
#include "aidge/utils/Types.h" #include "aidge/utils/Types.h"
namespace Aidge { namespace Aidge {
enum class ConcatAttr { NbInputs, Axis}; enum class ConcatAttr { NbInputs, Axis };
class Concat_Op : public Operator, class Concat_Op : public Operator,
public Registrable<Concat_Op, std::string, std::unique_ptr<OperatorImpl>(const Concat_Op&)>, 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 = ...@@ -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_ */ #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