From da7b49c245ee57b5552a93dc742968747ade19da Mon Sep 17 00:00:00 2001
From: Johan Gunnarsson <johan.gunnarsson@arm.com>
Date: Tue, 29 Apr 2025 11:16:35 +0200
Subject: [PATCH] MLBEDSW-10744: Fix DepthwiseConv2D with batch + depth
 multiplier

DepthwiseConv2d with batch and depth multiplier is decomposed into
multiple non-batched DepthwiseConv2d followed by Transpose, then a
MemoryCopy to copy the result into the right place in the original
OFM. This last MemoryCopy didn't have the correct IFM shape.

Signed-off-by: Johan Gunnarsson <johan.gunnarsson@arm.com>
Change-Id: I7fe23dbd0bbbb3aca6885153e76a2d3fc8ad7724
---
 ethosu/regor/compiler/scheduler_decompose.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ethosu/regor/compiler/scheduler_decompose.cpp b/ethosu/regor/compiler/scheduler_decompose.cpp
index ada19051..41167bbb 100644
--- a/ethosu/regor/compiler/scheduler_decompose.cpp
+++ b/ethosu/regor/compiler/scheduler_decompose.cpp
@@ -84,7 +84,7 @@ static std::unique_ptr<SchedulerOperation> MakeMemCopy(const std::shared_ptr<Sch
         ifmConn->tensor->storageShape = source->storageShape.WithDepth(2 * source->storageShape.Depth());
         source->RemoveReader(op.get());
     }
-    ifmConn->shape = ofmConn->shape;
+    ifmConn->shape = ofmConn->SliceShape();
 
     return op;
 }
-- 
GitLab