From ce6d5a01ba1b01461a3869a690b27fe417a0b374 Mon Sep 17 00:00:00 2001
From: hrouis <houssemeddine.rouis92@gmail.com>
Date: Thu, 7 Mar 2024 16:23:57 +0100
Subject: [PATCH] update gather and slice to be registrable
---
include/aidge/operator/Gather.hpp | 2 +-
include/aidge/operator/Slice.hpp | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/include/aidge/operator/Gather.hpp b/include/aidge/operator/Gather.hpp
index 43077b731..78557d49d 100644
--- a/include/aidge/operator/Gather.hpp
+++ b/include/aidge/operator/Gather.hpp
@@ -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:
diff --git a/include/aidge/operator/Slice.hpp b/include/aidge/operator/Slice.hpp
index e71eaf40f..186c60ffa 100644
--- a/include/aidge/operator/Slice.hpp
+++ b/include/aidge/operator/Slice.hpp
@@ -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;
--
GitLab