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

Fix issue that may be related to #142

parent 94b9cd98
No related branches found
No related tags found
2 merge requests!212Version 0.3.0,!161Add support for namespaced attributes in DynamicAttributes
Pipeline #50742 canceled
...@@ -78,6 +78,7 @@ bool Aidge::Split_Op::forwardDims(bool allowDataDependency) { ...@@ -78,6 +78,7 @@ bool Aidge::Split_Op::forwardDims(bool allowDataDependency) {
} }
std::shared_ptr<Tensor> fallback; std::shared_ptr<Tensor> fallback;
this->split().clear(); // If both are provided input would override attrs
this->split().reserve(getInput(1)->size()); this->split().reserve(getInput(1)->size());
const auto& splits = getInput(1)->refCastFrom(fallback, NativeType<DimSize_t>::type, "cpu"); const auto& splits = getInput(1)->refCastFrom(fallback, NativeType<DimSize_t>::type, "cpu");
std::copy_n(static_cast<DimSize_t*>(splits.getImpl()->hostPtr()), std::copy_n(static_cast<DimSize_t*>(splits.getImpl()->hostPtr()),
......
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