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

ADD: missing default values to ReduceMean constructor

parent a4a7401c
No related branches found
No related tags found
2 merge requests!318[Upd] release verision 0.5.0,!304UPD: remove 'iostream' and include only what is used
......@@ -52,12 +52,12 @@ public:
/**
* @brief constructor for ReduceMean op
* @param[in] axes around which perform the operation
* @param[in] keep_dims if true we set a dimension of 1 in the place of the reduced axes and
* @param[in] keep_dims if true we set a dimension of 1 in the place of the reduced axes and
* if false we remove the dimension completely
* @param[in] noop_with_empty_axes used when no axes are provided, if set to true, the operator does nothing
* and if false, we reduce on all axes
*/
ReduceMean_Op(const std::vector<std::int32_t>& axes, bool keep_dims, bool noop_with_empty_axes);
ReduceMean_Op(const std::vector<std::int32_t>& axes, bool keep_dims = true, bool noop_with_empty_axes = false);
/**
* @brief Copy-constructor. Copy the operator attributes and its output tensor(s), but not its input tensors (the new operator has no input associated).
......@@ -116,8 +116,8 @@ public:
// template <DimIdx_t DIM>
// const std::string ReduceMean_Op::Type = "ReduceMean";
std::shared_ptr<Node> ReduceMean(const std::vector<std::int32_t> &axes,
bool keep_dims=true,
bool noop_with_empty_axes=false,
bool keep_dims = true,
bool noop_with_empty_axes = false,
const std::string& name = "");
} // namespace Aidge
......
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