diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-05-28 16:59:27 +0100 |
---|---|---|
committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-05-29 12:13:05 +0000 |
commit | 97b2bb149862a928f936bade5c2154bb7d05bf82 (patch) | |
tree | 7eb7a3d73a3d6304543bb126b4808a27eaea5266 /tests | |
parent | c2a47163bcd50750ce8f864721ec0cb6e94b28e4 (diff) | |
download | ComputeLibrary-97b2bb149862a928f936bade5c2154bb7d05bf82.tar.gz |
COMPMID-2336: Reduce large and medium shapes
Change-Id: I464f97e908436bcc17622dca071096d93407618e
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1239
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'tests')
-rw-r--r-- | tests/datasets/ShapeDatasets.h | 17 |
1 files changed, 1 insertions, 16 deletions
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index 17b7a2c013..07ecf45d81 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -228,9 +228,6 @@ public: // Batch size 1 TensorShape{ 37U, 37U }, TensorShape{ 27U, 33U, 2U }, - TensorShape{ 128U, 64U, 1U, 3U }, - // Batch size 4 - TensorShape{ 37U, 37U, 3U, 4U }, // Arbitrary batch size TensorShape{ 37U, 37U, 3U, 5U } }) @@ -264,7 +261,6 @@ public: { TensorShape{ 42U, 37U, 8U }, TensorShape{ 57U, 60U, 13U }, - TensorShape{ 128U, 64U, 21U }, TensorShape{ 83U, 72U, 14U } }) { @@ -280,7 +276,6 @@ public: { TensorShape{ 42U, 37U, 8U, 15U }, TensorShape{ 57U, 60U, 13U, 8U }, - TensorShape{ 128U, 64U, 21U, 13U }, TensorShape{ 83U, 72U, 14U, 5U } }) { @@ -294,11 +289,7 @@ public: LargeShapes() : ShapeDataset("Shape", { - // Batch size 1 - TensorShape{ 1921U, 1083U }, - TensorShape{ 641U, 485U, 2U, 3U }, - // Batch size 4 - TensorShape{ 799U, 595U, 1U, 4U }, + TensorShape{ 582U, 131U, 1U, 4U }, }) { } @@ -695,12 +686,9 @@ public: : 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{ 33U, 35U, 8U, 8U } }) { } @@ -714,12 +702,9 @@ public: : 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{ 33U, 35U, 3U, 8U }, // Arbitrary batch size TensorShape{ 32U, 37U, 3U, 8U } }) |