diff --git a/ethosu/regor/compiler/graphir_optimiser.cpp b/ethosu/regor/compiler/graphir_optimiser.cpp index 2b44bd02ea2aeea00a27f7c9a68d64855f66c8ff..590bbc8a41ff91dea6504b71d915bfd2fec51440 100644 --- a/ethosu/regor/compiler/graphir_optimiser.cpp +++ b/ethosu/regor/compiler/graphir_optimiser.cpp @@ -959,7 +959,7 @@ Operation *GraphIrOptimiser::FuseRescale(Graph *const graph, Operation *const op { // Propagate rescaling to output of previous op producer->CopyOutput(TensorUsage::OFM, *ofmConn); - producer->Output(TensorUsage::OFM)->Set(ofmConn->rounding); + producer->Output(TensorUsage::OFM)->Set(ofmConn->rounding).Set(ofmQuant); returnOp = producer.get(); } }