From 3dd5b6884a65c06bcb9d15589ee2dc2978e3b336 Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Mon, 4 Mar 2019 14:14:02 +0000 Subject: COMPMID-1933: Implement NEHeightConcatenateLayer. Added support to concactenate tensors along the Y axis in NEConcatenateLayer. Change-Id: Ib714bfcf9954cc35918efa7d52fc9164bb08bdf6 Signed-off-by: Pablo Tello Reviewed-on: https://review.mlplatform.org/c/841 Reviewed-by: Gian Marco Iodice Reviewed-by: Georgios Pinitas Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins --- tests/datasets/ShapeDatasets.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'tests/datasets') diff --git a/tests/datasets/ShapeDatasets.h b/tests/datasets/ShapeDatasets.h index fddc7731fe..b6923c15c9 100644 --- a/tests/datasets/ShapeDatasets.h +++ b/tests/datasets/ShapeDatasets.h @@ -732,11 +732,11 @@ public: } }; -/** Data set containing tensor shapes for WidthConcatenateLayer. */ -class WidthConcatenateLayerShapes final : public ShapeDataset +/** Data set containing tensor shapes for ConcatenateLayer. */ +class ConcatenateLayerShapes final : public ShapeDataset { public: - WidthConcatenateLayerShapes() + ConcatenateLayerShapes() : ShapeDataset("Shape", { TensorShape{ 232U, 65U, 3U }, -- cgit v1.2.1