commit c55cef103a329a4e203c929276ed02ac45820de1
author Georgios Pinitas <georgios.pinitas@arm.com> Wed Aug 01 15:24:18 2018 +0100
committer Anthony Barbier <anthony.barbier@arm.com> Fri Nov 02 16:54:54 2018 +0000
tree e465b681254ea3182ad5257c10446265dc303def
parent ff4bbce113892ee18621bd95aae7f50db66257e0

COMPMID-1188: Expose FullyConnectedLayer info metadata at graph level

Change-Id: I7670f79209a1e4439d57e05c1f5c576f600971cb
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/142299
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>

arm_compute/graph/GraphBuilder.h

diff --git a/arm_compute/graph/GraphBuilder.h b/arm_compute/graph/GraphBuilder.h
index 5bb1df4..a2f7618 100644
--- a/arm_compute/graph/GraphBuilder.h
+++ b/arm_compute/graph/GraphBuilder.h
@@ -218,6 +218,7 @@
      * @param[in] num_outputs        Number of output neurons
      * @param[in] weights_accessor   (Optional) Accessor of the weights node data
      * @param[in] bias_accessor      (Optional) Accessor of the bias node data
+     * @param[in] fc_info            (Optional) Fully connected layer metadata
      * @param[in] weights_quant_info (Optional) Weights quantization info
      * @param[in] out_quant_info     (Optional) Output quantization info
      *
@@ -225,8 +226,9 @@
      */
     static NodeID add_fully_connected_layer(Graph &g, NodeParams params, NodeIdxPair input, unsigned int num_outputs,
                                             ITensorAccessorUPtr weights_accessor = nullptr, ITensorAccessorUPtr bias_accessor = nullptr,
-                                            const QuantizationInfo weights_quant_info = QuantizationInfo(),
-                                            const QuantizationInfo out_quant_info     = QuantizationInfo());
+                                            const FullyConnectedLayerInfo fc_info            = FullyConnectedLayerInfo(),
+                                            const QuantizationInfo        weights_quant_info = QuantizationInfo(),
+                                            const QuantizationInfo        out_quant_info     = QuantizationInfo());
     /** Adds a normalization layer node to the graph
      *
      * @param[in] g         Graph to add the node to