aboutsummaryrefslogtreecommitdiff
path: root/tests/datasets/FullyConnectedLayerDataset.h
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2017-10-20 18:52:20 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:35:24 +0000
commit96880cf00707d394938ec7fe31c21c79a2ac3f0c (patch)
treee7022f32813dd4df943b9e3b9d0948019d9b2af2 /tests/datasets/FullyConnectedLayerDataset.h
parent0a7a8d19d7c172a05402578b1e8ddd8ecb825643 (diff)
downloadComputeLibrary-96880cf00707d394938ec7fe31c21c79a2ac3f0c.tar.gz
COMPMID-640: FullyConnectedLayer failures on both NEON/CL
Change-Id: Idd830cff054114123229c189e423b753b8064146 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/92623 Reviewed-by: Robert Hughes <robert.hughes@arm.com> Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com> Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'tests/datasets/FullyConnectedLayerDataset.h')
-rw-r--r--tests/datasets/FullyConnectedLayerDataset.h6
1 files changed, 6 insertions, 0 deletions
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
@@ -120,10 +120,16 @@ 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));