commit 5fbe3e141c5c8301b0f8698b45b22a44705cfb11
author Miao Wang <miaowang@google.com> Tue Feb 20 14:34:00 2018 -0800
committer Miao Wang <miaowang@google.com> Tue Feb 20 14:44:08 2018 -0800
tree 33d55ea73ede341416c57139d4accc34c26fb969
parent f269b57b99f1350be793b221a392fd3aae14f17a

Fix a typo in spaceToBatchPrepare.

Bug: 73660399
Test: mm
Test: newly added SPACE_TO_BATCH tests pass
Change-Id: Ibf28243231e0122faa71960c56a759c616c43a41

common/OperationsUtils.cpp

diff --git a/common/OperationsUtils.cpp b/common/OperationsUtils.cpp
index 7ced5b2..816ee27 100644
--- a/common/OperationsUtils.cpp
+++ b/common/OperationsUtils.cpp
@@ -653,12 +653,12 @@
     uint32_t paddedWidth = paddingsData[2] + width + paddingsData[3];
 
     NN_OPS_CHECK(paddedHeight % blockSizeData[0] == 0);
-    NN_OPS_CHECK(paddedHeight % blockSizeData[1] == 0);
+    NN_OPS_CHECK(paddedWidth % blockSizeData[1] == 0);
 
     output->type = input.type;
     output->dimensions = {batches * (blockSizeData[0] * blockSizeData[1]),
                           paddedHeight / blockSizeData[0],
-                          paddedHeight / blockSizeData[1],
+                          paddedWidth / blockSizeData[1],
                           channels};
     output->offset = input.offset;
     output->scale = input.scale;