From 2219dea47ff74cadf2a6ee1ab95e57cd96e60596 Mon Sep 17 00:00:00 2001 From: zhenglin Date: Tue, 30 Jan 2018 18:15:52 +0800 Subject: 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 Reviewed-by: Joel Liang Reviewed-by: Anthony Barbier --- tests/datasets/FullyConnectedLayerDataset.h | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'tests/datasets/FullyConnectedLayerDataset.h') 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 * @@ -119,6 +119,8 @@ class SmallFullyConnectedLayerDataset final : public FullyConnectedLayerDataset 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 -- cgit v1.2.1