Skip to content
Snippets Groups Projects
Commit 87ce7424 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Added missing clone for new MaxPooling_Op

parent 55825145
No related branches found
No related tags found
1 merge request!8GraphView cloning proposal + labelGraph proof of concept
Pipeline #31894 passed
...@@ -63,6 +63,28 @@ public: ...@@ -63,6 +63,28 @@ public:
setDatatype(DataType::Float32); setDatatype(DataType::Float32);
} }
/**
* @brief Copy-constructor. Copy the operator parameters and its output tensor(s), but not its input tensors (the new operator has no input associated).
* @param op Operator to copy.
*/
MaxPooling_Op(const MaxPooling_Op& op)
: Operator(Type),
Parameterizable_(op),
mOutput(std::make_shared<Tensor>(*op.mOutput))
{
// cpy-ctor
setDatatype(op.mOutput->dataType());
}
/**
* @brief Clone the operator using its copy-constructor.
* @see Operator::MaxPooling_Op
* @param op Operator to copy.
*/
Operator* clone() const override {
return new MaxPooling_Op<DIM>(*static_cast<const MaxPooling_Op<DIM>*>(this));
}
constexpr void associateInput(const IOIndex_t inputIdx, std::shared_ptr<Data> data) override final { constexpr void associateInput(const IOIndex_t inputIdx, std::shared_ptr<Data> data) override final {
assert(inputIdx < 1 && "operators supports only 3 inputs"); assert(inputIdx < 1 && "operators supports only 3 inputs");
(void) inputIdx; // avoid unused warning (void) inputIdx; // avoid unused warning
......
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