Skip to content
Snippets Groups Projects

[Add] Resize Operator unit-tests

Closed Michal Szczepanski requested to merge mszczep/aidge_core:operator_resize into dev
All threads resolved!
1 file
+ 3
2
Compare changes
  • Side-by-side
  • Inline
+ 3
2
@@ -75,10 +75,11 @@ bool Aidge::Resize_Op::forwardDims(bool allowDataDependency) {
for (std::size_t dim=0; dim < getInput(2)->size(); ++dim) {
float scale = static_cast<float*>(scales.getImpl()->hostPtr())[dim];
outDims[dim] = inDims[dim] * static_cast<int64_t>(scale);
fmt::print("scale {} = {} \n",dim, scale);
outDims[dim] = static_cast<int64_t>(inDims[dim] * scale);
// outDims[dim] = inDims[dim]*static_cast<int64_t*>(scales.getImpl()->hostPtr())[dim];
}
fmt::print("OutDims = {} \n", outDims);
mOutputs[0]->resize(outDims);
return true;
}
Loading