diff options
| -rw-r--r-- | tests/datasets_new/ShapeDatasets.h | 10 |
1 files changed, 2 insertions, 8 deletions
diff --git a/tests/datasets_new/ShapeDatasets.h b/tests/datasets_new/ShapeDatasets.h index a693be7153..72681d7814 100644 --- a/tests/datasets_new/ShapeDatasets.h +++ b/tests/datasets_new/ShapeDatasets.h @@ -97,15 +97,9 @@ public: TensorShape{ 1245U, 652U, 1U, 3U }, TensorShape{ 4160U, 3120U }, // Batch size 4 - TensorShape{ 1920U, 1080U, 4U }, - TensorShape{ 1245U, 652U, 1U, 3U, 4U }, - TensorShape{ 4160U, 3120U, 4U }, + TensorShape{ 1245U, 652U, 1U, 4U }, // Batch size 8 - TensorShape{ 1920U, 1080U, 8U }, - TensorShape{ 1245U, 652U, 1U, 3U, 8U }, - TensorShape{ 4160U, 3120U, 8U }, - // Arbitrary batch size - TensorShape{ 1920U, 1080U, 5U }, + TensorShape{ 1245U, 652U, 1U, 8U }, }) { } |