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
3 merge requests!105version 0.2.0,!88Basic supervised learning,!82Resolve "Optimizer to update gradients"
Pipeline #41897 failed
......@@ -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