From bbbeb0300e2511d8e824d436ef1488129f967668 Mon Sep 17 00:00:00 2001 From: Fredrik Svedberg Date: Wed, 30 Apr 2025 13:22:53 +0200 Subject: [PATCH] MLBEDSW-9760 TOSA Conv3D Decomposition Final part of TOSA Conv3D decomposition - bias broadcast decomposition of large tensors. Change-Id: Ib2e278266f431fbfddffd7aef850a32fd63c3e17 Signed-off-by: Fredrik Svedberg --- ethosu/regor/compiler/scheduler_decompose.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ethosu/regor/compiler/scheduler_decompose.cpp b/ethosu/regor/compiler/scheduler_decompose.cpp index 81527ea4..c3c74440 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 ) { -- GitLab