diff options
Diffstat (limited to 'tests/dataset')
-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 ecb478dbf0..1e9e5f139a 100644 --- a/tests/dataset/ShapeDatasets.h +++ b/tests/dataset/ShapeDatasets.h @@ -97,7 +97,7 @@ class Small1DShape final : public ShapeDataset<1> { public: Small1DShape() - : ShapeDataset(TensorShape(128U)) + : ShapeDataset(TensorShape(256U)) { } }; |