diff --git a/ethosu/regor/compiler/scheduler_decompose.cpp b/ethosu/regor/compiler/scheduler_decompose.cpp
index 9dafb6ce9f268b3b30add7192667f21bdfaaeaa5..096bae9a5c1d7db08aed3d701ca69b9fa2055977 100644
--- a/ethosu/regor/compiler/scheduler_decompose.cpp
+++ b/ethosu/regor/compiler/scheduler_decompose.cpp
@@ -433,7 +433,7 @@ HandleDilation(Architecture *arch, std::unique_ptr<SchedulerOperation> op, const
                 newOfmSlice.offset.WithHeight(newOfmSlice.offset.Height() + dy).WithWidth(newOfmSlice.offset.Width() + dx);
             newOfmSlice.shape =
                 newOfmSlice.shape.WithHeight(newOfmSlice.shape.Height() - dy).WithWidth(newOfmSlice.shape.Width() - dx);
-            if ( newOfmSlice.shape.Elements() > 0 )
+            if ( newOfmSlice.shape.Width() > 0 && newOfmSlice.shape.Height() > 0 )
             {
                 ofmStrides.y *= DY;
                 ofmStrides.x *= DX;