APPBROWSER-400: Implement the tensorshift kernel for fixing DC's alignment issue on OpenGL ES
Change-Id: I7a8489bb0fddc72899ea165e414ee87bdbfb45b3
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/118106
Reviewed-by: Joel Liang <joel.liang@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h
index dbcd9d5..9114f51 100644
--- a/tests/datasets/ShapeDatasets.h
+++ b/tests/datasets/ShapeDatasets.h
@@ -387,6 +387,28 @@
}
};
+/** Data set containing small tensor shapes for direct convolution. */
+class SmallDirectConvolutionTensorShiftShapes final : public ShapeDataset
+{
+public:
+ SmallDirectConvolutionTensorShiftShapes()
+ : ShapeDataset("InputShape",
+ {
+ // Batch size 1
+ TensorShape{ 35U, 35U, 3U },
+ TensorShape{ 32U, 37U, 3U },
+ // Batch size 4
+ TensorShape{ 32U, 37U, 3U, 4U },
+ // Batch size 8
+ TensorShape{ 32U, 37U, 3U, 8U },
+ TensorShape{ 33U, 35U, 3U, 8U },
+ // Arbitrary batch size
+ TensorShape{ 32U, 37U, 3U, 8U }
+ })
+ {
+ }
+};
+
/** Data set containing 2D tensor shapes for DepthConcatenateLayer. */
class DepthConcatenateLayerShapes final : public ShapeDataset
{