2 files changed, 23 insertions, 1 deletions
diff --git a/tests/datasets/LargeMatMulDataset.h b/tests/datasets/LargeMatMulDataset.h
index b5181bc30b..8f6c000d37 100644
--- a/tests/datasets/LargeMatMulDataset.h
+++ b/tests/datasets/LargeMatMulDataset.h
@@ -65,7 +65,17 @@ public:
         add_config(TensorShape(45U, 38U, 3U, 2U, 3U), TensorShape(20U, 45U, 3U, 2U, 3U), TensorShape(20U, 38U, 3U, 2U, 3U));
     }
 };
-
+class LargeMatMulDatasetRhsExportToCLImageRhsT final : public MatMulDataset
+{
+public:
+    // For shape choices, please refer to the explanations given in SmallMatMulDatasetRhsExportToCLImageRhsT
+    LargeMatMulDatasetRhsExportToCLImageRhsT()
+    {
+        add_config(TensorShape(28U, 13U, 3U, 2U), TensorShape(32U, 28U, 3U, 2U), TensorShape(32U, 13U, 3U, 2U));
+        add_config(TensorShape(40U, 12U, 1U, 5U, 2U), TensorShape(20U, 40U, 1U, 5U, 2U), TensorShape(20U, 12U, 1U, 5U, 2U));
+        add_config(TensorShape(44U, 38U, 3U, 2U, 3U), TensorShape(20U, 44U, 3U, 2U, 3U), TensorShape(20U, 38U, 3U, 2U, 3U));
+    }
+};
 } // namespace datasets
 } // namespace test
 } // namespace arm_compute
diff --git a/tests/datasets/SmallMatMulDataset.h b/tests/datasets/SmallMatMulDataset.h
index 93e5f7dc2c..52ef01da7b 100644
--- a/tests/datasets/SmallMatMulDataset.h
+++ b/tests/datasets/SmallMatMulDataset.h
@@ -57,6 +57,18 @@ public:
     }
 };
 
+class SmallMatMulDatasetRhsExportToCLImageRhsT final : public MatMulDataset
+{
+public:
+    // Some considerations:
+    //  1) K dimension should be a multiple of 4
+    //  See (2), (3), and (4) in SmallMatMulDatasetRhsExportToCLImageRhsNT
+    SmallMatMulDatasetRhsExportToCLImageRhsT()
+    {
+        add_config(TensorShape(8U /*K*/, 3U /*M*/, 2U, 1U, 2U), TensorShape(20U /*N*/, 8U /*K*/, 2U, 1U, 2U), TensorShape(20U /*N*/, 3U /*M*/, 2U, 1U, 2U));
+    }
+};
+
 class SmallMatMulDatasetRhsExportToCLImageRhsNT final : public MatMulDataset
 {
 public: