From 341b218eff8c75aa9862b333bb0012a03c14a1ef Mon Sep 17 00:00:00 2001 From: John Kesapides Date: Fri, 22 Feb 2019 10:05:29 +0000 Subject: COMPMID-1493 Create tests/validate_examples/graph_fully_connected Add graph example with validation for fully-connected layer Change-Id: I06fcc670b7097609f04eb040fedf56108c9484d2 Signed-off-by: John Kesapides Reviewed-on: https://review.mlplatform.org/c/764 Tested-by: Arm Jenkins Reviewed-by: Pablo Marquez Reviewed-by: Georgios Pinitas Comments-Addressed: Arm Jenkins --- tests/validation/reference/FullyConnectedLayer.h | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'tests/validation/reference/FullyConnectedLayer.h') diff --git a/tests/validation/reference/FullyConnectedLayer.h b/tests/validation/reference/FullyConnectedLayer.h index 1dfb496924..f474a1cfb8 100644 --- a/tests/validation/reference/FullyConnectedLayer.h +++ b/tests/validation/reference/FullyConnectedLayer.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -36,7 +36,8 @@ namespace validation namespace reference { template -SimpleTensor fully_connected_layer(const SimpleTensor &src, const SimpleTensor &weights, const SimpleTensor &bias, const TensorShape &dst_shape); +SimpleTensor fully_connected_layer(const SimpleTensor &src, const SimpleTensor &weights, const SimpleTensor &bias, const TensorShape &dst_shape, + QuantizationInfo out_quant_info = QuantizationInfo()); } // namespace reference } // namespace validation } // namespace test -- cgit v1.2.1