diff options
Diffstat (limited to 'tests/datasets')
-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 bd29fe649a..9ee89f43e0 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -148,7 +148,7 @@ public: Small4DShapes() : ShapeDataset("Shape", { - TensorShape{ 1U, 7U, 1U, 3U }, + TensorShape{ 2U, 7U, 1U, 3U }, TensorShape{ 7U, 7U, 5U, 3U }, TensorShape{ 27U, 13U, 37U, 2U }, TensorShape{ 128U, 64U, 21U, 3U } |