Skip to content
Snippets Groups Projects
Commit ce6d5a01 authored by Houssem ROUIS's avatar Houssem ROUIS
Browse files

update gather and slice to be registrable

parent ade64013
No related branches found
No related tags found
No related merge requests found
......@@ -32,7 +32,7 @@ enum class GatherAttr { Axis };
class Gather_Op : public OperatorTensor,
public Registrable<Gather_Op,
std::string,
std::unique_ptr<OperatorImpl>(const Gather_Op&)>,
std::shared_ptr<OperatorImpl>(const Gather_Op&)>,
public StaticAttributes<GatherAttr, std::int64_t> {
public:
......
......@@ -25,7 +25,7 @@
namespace Aidge {
class Slice_Op
: public OperatorTensor,
public Registrable<Slice_Op, std::string, std::unique_ptr<OperatorImpl>(const Slice_Op &)>{
public Registrable<Slice_Op, std::string, std::shared_ptr<OperatorImpl>(const Slice_Op &)>{
public:
static const std::string Type;
......
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