diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-10-20 18:52:20 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | 96880cf00707d394938ec7fe31c21c79a2ac3f0c (patch) | |
tree | e7022f32813dd4df943b9e3b9d0948019d9b2af2 /tests/datasets/FullyConnectedLayerDataset.h | |
parent | 0a7a8d19d7c172a05402578b1e8ddd8ecb825643 (diff) | |
download | ComputeLibrary-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.h | 6 |
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)); |