Skip to content
Snippets Groups Projects

[Fix] Dynamically set output Tensor feature of `Identity_Op` based on input Tensor

Merged Olivier BICHLER requested to merge fix_identity into dev
1 file
+ 3
0
Compare changes
  • Side-by-side
  • Inline
+ 3
0
@@ -15,6 +15,9 @@
void Aidge::Identity_OpImpl::forward() {
const Identity_Op& op = dynamic_cast<const Identity_Op&>(mOp);
op.getOutput(0)->setBackend(op.getInput(0)->backend(), op.getInput(0)->device());
op.getOutput(0)->setDataType(op.getInput(0)->dataType());
op.getOutput(0)->setDataFormat(op.getInput(0)->dataFormat());
op.getOutput(0)->getImpl()->copy(op.getInput(0)->getImpl()->rawPtr(), op.getInput(0)->size());
}
Loading