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

Fix OperatorTensor.cpp

parent 00842c0d
No related branches found
No related tags found
No related merge requests found
......@@ -91,7 +91,7 @@ void Aidge::OperatorTensor::setOutput(const Aidge::IOIndex_t outputIdx, const st
void Aidge::OperatorTensor::setOutput(const Aidge::IOIndex_t outputIdx, std::shared_ptr<Aidge::Data>&& data) {
AIDGE_ASSERT(data->type() == Tensor::Type, "{} Operator only accepts Tensors as inputs", type());
AIDGE_ASSERT(outputIdx < nbOutputs(), "{} Operator has {} outputs", type(), nbOutputs());
auto&& data_tensor =std::move(std::dynamic_pointer_cast<Tensor>(data));
auto&& data_tensor = std::dynamic_pointer_cast<Tensor>(data);
// if (mImpl)
// AIDGE_ASSERT(data_tensor->getImpl()->backend() == backend(), "Data parameter and Operator have different backends: {} and {}", data_tensor->getImpl()->backend(), backend());
*mOutputs[outputIdx] = std::move(*data_tensor);
......
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