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 d2b82cae40..ecb478dbf0 100644 --- a/tests/dataset/ShapeDatasets.h +++ b/tests/dataset/ShapeDatasets.h @@ -118,7 +118,7 @@ class SmallShapes final : public ShapeDataset<3> { public: SmallShapes() - : ShapeDataset(TensorShape(5U, 5U), + : ShapeDataset(TensorShape(7U, 7U), TensorShape(27U, 13U, 2U), TensorShape(128U, 64U, 1U, 3U)) { |