diff options
Diffstat (limited to 'tests/datasets/ShapeDatasets.h')
-rw-r--r-- | tests/datasets/ShapeDatasets.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index 173ee74958..02a71aa7b5 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -269,11 +269,11 @@ public: } }; -/** Data set containing 2D tensor shapes for DepthConcatenate. */ -class DepthConcatenateShapes final : public ShapeDataset +/** Data set containing 2D tensor shapes for DepthConcatenateLayer. */ +class DepthConcatenateLayerShapes final : public ShapeDataset { public: - DepthConcatenateShapes() + DepthConcatenateLayerShapes() : ShapeDataset("Shape", { TensorShape{ 322U, 243U }, |