diff --git a/ethosu/regor/architecture/ethosu85/ethos_u85_register_cs_generator.cpp b/ethosu/regor/architecture/ethosu85/ethos_u85_register_cs_generator.cpp index 0ab1724684b0644af2b1828ac5095b5e0166b18f..607eac86c4f94441eabb7f1b5c2cb35f196f2fbd 100644 --- a/ethosu/regor/architecture/ethosu85/ethos_u85_register_cs_generator.cpp +++ b/ethosu/regor/architecture/ethosu85/ethos_u85_register_cs_generator.cpp @@ -692,7 +692,7 @@ void EthosU85RCSGenerator::GenerateOFMScalingForPooling(HLCOperation *poolOp, bo QuantizedScale ofmScale(1, 0); pooling_mode mode = (poolOp->type == OpType::AvgPool && (poolOp->kernel.Size().x > 8 || poolOp->kernel.Size().y > 8)) ? pooling_mode::SUM : pooling_mode::NONE; - if ( mode == pooling_mode::SUM && useGlobalScale && poolOp->kernel.Padding().IsZero() ) + if ( mode == pooling_mode::SUM && useGlobalScale && !poolOp->ofm.quantization.scales.empty() ) { uint32_t scale = 1; int shift = 0;