Simplify shape checking for TRANSPOSE

Have a simple outer check for proper size rather than
checking during the inner loop.

Change-Id: I20312c5f6b4aacbb630a2df0f901d8586caa04c0
Signed-off-by: Eric Kunze <eric.kunze@arm.com>

1 files changed, 8 insertions, 2 deletions

diff --git a/chapters/data_layout.adoc b/chapters/data_layout.adoc
index e50a14e..395843a 100644
--- a/chapters/data_layout.adoc
+++ b/chapters/data_layout.adoc
@@ -309,11 +309,17 @@ for_each(index in perms) {
     ERROR_IF(indexes_used[index] == true);
     indexes_used[index] = true;
 }
+
+// Ensure that the output shapes have the properly
+// permuted shapes
+for(i = 0; i < rank(shape); i++) {
+    ERROR_IF(shape1[perms[i]] != shape[i])
+}
+
 for_each(index in shape) {
     tmp_index = index;
     for(i = 0; i < rank(shape); i++) {
-        REQUIRE(shape1[perm[i]] == shape[i])
-        tmp_index[perm[i]] = index[i]
+        tmp_index[perms[i]] = index[i]
     }
     in_out_t value = tensor_read<in_out_t>(input, shape1, tmp_index);
     tensor_write<in_out_t>(output, shape, index, value);