COMPMID-1048 Add NHWC data format support to Winograd filter transform 4x4_3x3
Change-Id: Ifd125fcb5451dbac3c28b15a9471048a74fee0ad
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/128987
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
diff --git a/src/core/Helpers.cpp b/src/core/Helpers.cpp
index e336331..c0af3bb 100644
--- a/src/core/Helpers.cpp
+++ b/src/core/Helpers.cpp
@@ -59,6 +59,13 @@
++n;
}
+ if(anchor.num_dimensions() > 2)
+ {
+ window.set(2, Window::Dimension(anchor[2], std::max<size_t>(1, shape[2]), steps[2]));
+
+ ++n;
+ }
+
for(; n < anchor.num_dimensions(); ++n)
{
window.set(n, Window::Dimension(anchor[n], std::max<size_t>(1, shape[n])));