Update reference generator test set calculation

Signed-off-by: Won Jeon <won.jeon@arm.com>
Change-Id: I74f9ac096e5030a86bd7c1ffff542d9061b09ab4
diff --git a/reference_model/src/generate/generate_dot_product_states.cc b/reference_model/src/generate/generate_dot_product_states.cc
index f372bde..fb4d1dd 100644
--- a/reference_model/src/generate/generate_dot_product_states.cc
+++ b/reference_model/src/generate/generate_dot_product_states.cc
@@ -86,8 +86,8 @@
 public:
     GeneratorS0(uint32_t p)
         : _p(p)
-        , _set_data0(2 * 0)
-        , _set_data1(2 * 0 + 1)
+        , _set_data0(3 * 0)
+        , _set_data1(3 * 0 + 1)
     {}
     float operator()(uint32_t k) override
     {
@@ -157,7 +157,7 @@
     GeneratorS2(uint32_t p, uint32_t KS)
         : _p(p)
         , _KS(KS)
-        , _set_data(2 * 2 + p)
+        , _set_data(3 * 2 + p)
     {}
     float operator()(uint32_t k) override
     {
@@ -188,7 +188,7 @@
 public:
     GeneratorS3(uint32_t p)
         : _p(p)
-        , _set_data(2 * 3 + p)
+        , _set_data(3 * 3 + p)
     {}
     float operator()(uint32_t k) override
     {
@@ -225,8 +225,8 @@
         : _p(p)
         , _KS(KS)
         , _B(B)
-        , _set_data0(2 * 4 + 0)
-        , _set_data1(2 * 4 + 1)
+        , _set_data0(3 * 4 + 0)
+        , _set_data1(3 * 4 + 1)
     {}
     float operator()(uint32_t k) override
     {