aboutsummaryrefslogtreecommitdiff
path: root/tests/datasets/ShapeDatasets.h
diff options
context:
space:
mode:
Diffstat (limited to 'tests/datasets/ShapeDatasets.h')
-rw-r--r--tests/datasets/ShapeDatasets.h3
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h
index 928ff736bc..8255672430 100644
--- a/tests/datasets/ShapeDatasets.h
+++ b/tests/datasets/ShapeDatasets.h
@@ -232,7 +232,8 @@ public:
TensorShape{ 42U, 37U },
TensorShape{ 57U, 60U },
TensorShape{ 128U, 64U },
- TensorShape{ 83U, 72U }
+ TensorShape{ 83U, 72U },
+ TensorShape{ 40U, 40U }
})
{
}