diff options
Diffstat (limited to 'tests/dataset/ShapeDatasets.h')
| -rw-r--r-- | tests/dataset/ShapeDatasets.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/dataset/ShapeDatasets.h b/tests/dataset/ShapeDatasets.h index 71c786a641..ccdfc9a7dd 100644 --- a/tests/dataset/ShapeDatasets.h +++ b/tests/dataset/ShapeDatasets.h @@ -107,7 +107,7 @@ class Small2DShapes final : public ShapeDataset<2> { public: Small2DShapes() - : ShapeDataset(TensorShape(5U, 5U), + : ShapeDataset(TensorShape(17U, 17U), TensorShape(640U, 480U)) { } |