commit 2f1366a931a8641d0f8c4ce28dc85dfa818236ed
author Georgios Pinitas <georgios.pinitas@arm.com> Tue Jul 31 16:33:06 2018 +0100
committer Anthony Barbier <anthony.barbier@arm.com> Fri Nov 02 16:54:54 2018 +0000
tree cf864a49de3f35fcce0682a7d2be5ed7d4e34a6f
parent ea0147d9b90a0ee24c42c129a5ebdc7ef3c85876

COMPMID-1188: Add quantization info support in graph FC layer.

Change-Id: Ie9a6a896da142198243139fb9f8be0f83b87ccce
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/142130
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Vidhya Sudhan Loganathan <vidhyasudhan.loganathan@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>

arm_compute/graph/nodes/FullyConnectedLayerNode.h

diff --git a/arm_compute/graph/nodes/FullyConnectedLayerNode.h b/arm_compute/graph/nodes/FullyConnectedLayerNode.h
index 79201c8..1bff600 100644
--- a/arm_compute/graph/nodes/FullyConnectedLayerNode.h
+++ b/arm_compute/graph/nodes/FullyConnectedLayerNode.h
@@ -36,30 +36,39 @@
 public:
     /** Constructor
      *
-     * @param[in] num_outputs Number of neurons in the layer
-     * @param[in] fc_info     (Optional) Additional information about the fully connected layer
+     * @param[in] num_outputs    Number of neurons in the layer
+     * @param[in] out_quant_info (Optional) Output quantization info
+     * @param[in] fc_info        (Optional) Additional information about the fully connected layer
      */
-    FullyConnectedLayerNode(unsigned int num_outputs, FullyConnectedLayerInfo fc_info = FullyConnectedLayerInfo());
+    FullyConnectedLayerNode(unsigned int            num_outputs,
+                            QuantizationInfo        out_quant_info = QuantizationInfo(),
+                            FullyConnectedLayerInfo fc_info        = FullyConnectedLayerInfo());
     /** Computes weights descriptor
      *
      * @warning Works for inputs with 1D batch space
      *
-     * @param[in] input_descriptor Input descriptor
-     * @param[in] num_outputs      Number of output neurons
+     * @param[in] input_descriptor   Input descriptor
+     * @param[in] num_outputs        Number of output neurons
+     * @param[in] weights_quant_info (Optional) Weights quantization info
      *
      * @return Weights descriptor
      */
-    static TensorDescriptor compute_weights_descriptor(const TensorDescriptor &input_descriptor, unsigned int num_outputs);
+    static TensorDescriptor compute_weights_descriptor(const TensorDescriptor &input_descriptor,
+                                                       unsigned int            num_outputs,
+                                                       QuantizationInfo        weights_quant_info = QuantizationInfo());
     /** Computes fully connected layer output descriptor
      *
      * @warning Works for inputs with 1D batch space
      *
      * @param[in] input_descriptor Input descriptor
      * @param[in] num_outputs      Number of output neurons
+     * @param[in] out_quant_info   (Optional) Weights quantization info
      *
      * @return Output descriptor
      */
-    static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor, unsigned int num_outputs);
+    static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor,
+                                                      unsigned int            num_outputs,
+                                                      QuantizationInfo        out_quant_info = QuantizationInfo());
     /** Fully connected layer addition information
      *
      * @return Additional information about the fully connected layer
@@ -74,6 +83,7 @@
 
 private:
     unsigned int            _num_outputs;
+    QuantizationInfo        _out_quant_info;
     FullyConnectedLayerInfo _info;
 };
 } // namespace graph