Skip to content
Snippets Groups Projects

Fix Reshape

Merged Houssem ROUIS requested to merge hrouis/aidge_core:fix_reshape into dev
Files
4
@@ -45,7 +45,7 @@ public:
@@ -45,7 +45,7 @@ public:
using attr = typename Attributes_::template attr<e>;
using attr = typename Attributes_::template attr<e>;
Reshape_Op(const std::vector<std::int64_t>& shape)
Reshape_Op(const std::vector<std::int64_t>& shape)
: OperatorTensor(Type, 1, 0, 1),
: OperatorTensor(Type, 2, 0, 1),
Attributes_(attr<ReshapeAttr::Shape>(shape))
Attributes_(attr<ReshapeAttr::Shape>(shape))
{
{
mImpl = std::make_shared<Reshape_OpImpl>(*this);
mImpl = std::make_shared<Reshape_OpImpl>(*this);
@@ -87,7 +87,7 @@ public:
@@ -87,7 +87,7 @@ public:
}
}
};
};
inline std::shared_ptr<Node> Reshape(const std::vector<std::int64_t>& shape,
inline std::shared_ptr<Node> Reshape(const std::vector<std::int64_t>& shape = {},
const std::string &name = "") {
const std::string &name = "") {
// FIXME: properly handle default w&b initialization in every cases
// FIXME: properly handle default w&b initialization in every cases
return std::make_shared<Node>(std::make_shared<Reshape_Op>(shape), name);
return std::make_shared<Node>(std::make_shared<Reshape_Op>(shape), name);
Loading