diff --git a/ethosu/regor/compiler/scheduler_decompose.cpp b/ethosu/regor/compiler/scheduler_decompose.cpp index 0b5ba748c410b720f9bb1b2f8dab4ec04b83b83b..ada19051d5166f473daa23fa4ca02db37d5a78e4 100644 --- a/ethosu/regor/compiler/scheduler_decompose.cpp +++ b/ethosu/regor/compiler/scheduler_decompose.cpp @@ -1093,6 +1093,7 @@ std::vector> DecomposeDepthwiseConv2D(Archit subOfmConn->shape = transposedOfm->storageShape; subOfmConn->slice.offset = ofmShape.WithZeros().WithBatch(multiplier); subOfmConn->slice.shape = ofmShape.WithDepth(subOfmDepth); + subOfmConn->quantization = SliceQ(subOfmConn->quantization, multiplier, depthMultiplier); auto subOps = DecomposeDepthwiseConv2D(arch, std::move(subOp)); result.insert(result.end(), std::make_move_iterator(subOps.begin()), std::make_move_iterator(subOps.end())); }