diff options
Diffstat (limited to 'tests/datasets/ShapeDatasets.h')
-rw-r--r-- | tests/datasets/ShapeDatasets.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index f8c0309d04..a7f1a44286 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -683,7 +683,7 @@ public: // Batch size 1 TensorShape{ 32U, 37U, 3U }, // Batch size 4 - TensorShape{ 32U, 37U, 3U, 4U }, + TensorShape{ 6U, 9U, 5U, 4U }, }) { } |