From 96880cf00707d394938ec7fe31c21c79a2ac3f0c Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Fri, 20 Oct 2017 18:52:20 +0100 Subject: COMPMID-640: FullyConnectedLayer failures on both NEON/CL Change-Id: Idd830cff054114123229c189e423b753b8064146 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/92623 Reviewed-by: Robert Hughes Tested-by: Kaizen Reviewed-by: Anthony Barbier --- tests/datasets/FullyConnectedLayerDataset.h | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'tests/datasets/FullyConnectedLayerDataset.h') diff --git a/tests/datasets/FullyConnectedLayerDataset.h b/tests/datasets/FullyConnectedLayerDataset.h index 5789954e85..9f8089d81a 100644 --- a/tests/datasets/FullyConnectedLayerDataset.h +++ b/tests/datasets/FullyConnectedLayerDataset.h @@ -119,11 +119,17 @@ class SmallFullyConnectedLayerDataset final : public FullyConnectedLayerDataset public: SmallFullyConnectedLayerDataset() { + // Conv -> FC + add_config(TensorShape(1U, 1U, 1U, 3U), TensorShape(1U, 10U), TensorShape(10U), TensorShape(10U, 3U)); // Conv -> FC add_config(TensorShape(9U, 5U, 7U), TensorShape(315U, 271U), TensorShape(271U), TensorShape(271U)); // Conv -> FC (batched) add_config(TensorShape(9U, 5U, 7U, 3U), TensorShape(315U, 271U), TensorShape(271U), TensorShape(271U, 3U)); // FC -> FC + add_config(TensorShape(1U), TensorShape(1U, 10U), TensorShape(10U), TensorShape(10U)); + // FC -> FC (batched) + add_config(TensorShape(1U, 3U), TensorShape(1U, 10U), TensorShape(10U), TensorShape(10U, 3U)); + // FC -> FC add_config(TensorShape(201U), TensorShape(201U, 529U), TensorShape(529U), TensorShape(529U)); // FC -> FC (batched) add_config(TensorShape(201U, 3U), TensorShape(201U, 529U), TensorShape(529U), TensorShape(529U, 3U)); -- cgit v1.2.1