diff --git a/ethosu/regor/compiler/scheduler_decompose.cpp b/ethosu/regor/compiler/scheduler_decompose.cpp index 81527ea432bd17759b845e813bc256030dfe4557..c3c744400c7223b3f663128ccbdc52089b4b3d3b 100644 --- a/ethosu/regor/compiler/scheduler_decompose.cpp +++ b/ethosu/regor/compiler/scheduler_decompose.cpp @@ -957,8 +957,8 @@ std::vector> DecomposeConv3D(Architecture *a subOpIfm->slice.offset = ifm0shape.WithZeros(); subOpIfm->slice.shape = ifm0shape; - // TODO: MLBEDSW-9759 Pooling Decomposition - result.emplace_back(std::move(subOp)); + auto subOps = DecomposeElementwise(arch, std::move(subOp)); + result.insert(result.end(), std::make_move_iterator(subOps.begin()), std::make_move_iterator(subOps.end())); } else if ( conv2dSubOps.size() > 1 ) {