aboutsummaryrefslogtreecommitdiff
path: root/tests/datasets/FullyConnectedLayerDataset.h
diff options
context:
space:
mode:
authorzhenglin <zhenglin.li@arm.com>2018-01-30 18:15:52 +0800
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:45:00 +0000
commit2219dea47ff74cadf2a6ee1ab95e57cd96e60596 (patch)
tree7c74a3353ce3df529c77caf6052b2c2a768e6274 /tests/datasets/FullyConnectedLayerDataset.h
parent11c3b33215225f5baf34c045a68982e0058af74a (diff)
downloadComputeLibrary-2219dea47ff74cadf2a6ee1ab95e57cd96e60596.tar.gz
APPBROWSER-390,397,398: bugfix and fully connected validation issue on specific dataset
Change-Id: I227e90445715c3bd394e49930b010c0a5f5ca177 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/118108 Tested-by: Jenkins <bsgcomp@arm.com> Reviewed-by: Joel Liang <joel.liang@arm.com> Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'tests/datasets/FullyConnectedLayerDataset.h')
-rw-r--r--tests/datasets/FullyConnectedLayerDataset.h4
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/datasets/FullyConnectedLayerDataset.h b/tests/datasets/FullyConnectedLayerDataset.h
index b2008d604b..60183f8746 100644
--- a/tests/datasets/FullyConnectedLayerDataset.h
+++ b/tests/datasets/FullyConnectedLayerDataset.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2017 ARM Limited.
+ * Copyright (c) 2017-2018 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -120,6 +120,8 @@ public:
SmallFullyConnectedLayerDataset()
{
// Conv -> FC
+ add_config(TensorShape(8U, 1U, 1U), TensorShape(8U, 16U), TensorShape(16U), TensorShape(16U));
+ // 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));