COMPMID-1386: Add support for converting weights for CL.
Change-Id: I62e3ead903366baeeb1488f233a9b8b0c388c9de
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/140403
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
diff --git a/arm_compute/graph/nodes/FullyConnectedLayerNode.h b/arm_compute/graph/nodes/FullyConnectedLayerNode.h
index 3d1b689..79201c8 100644
--- a/arm_compute/graph/nodes/FullyConnectedLayerNode.h
+++ b/arm_compute/graph/nodes/FullyConnectedLayerNode.h
@@ -37,8 +37,9 @@
/** Constructor
*
* @param[in] num_outputs Number of neurons in the layer
+ * @param[in] fc_info (Optional) Additional information about the fully connected layer
*/
- FullyConnectedLayerNode(unsigned int num_outputs);
+ FullyConnectedLayerNode(unsigned int num_outputs, FullyConnectedLayerInfo fc_info = FullyConnectedLayerInfo());
/** Computes weights descriptor
*
* @warning Works for inputs with 1D batch space
@@ -59,6 +60,11 @@
* @return Output descriptor
*/
static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor, unsigned int num_outputs);
+ /** Fully connected layer addition information
+ *
+ * @return Additional information about the fully connected layer
+ */
+ FullyConnectedLayerInfo info() const;
// Inherited overridden methods:
NodeType type() const override;
@@ -67,7 +73,8 @@
void accept(INodeVisitor &v) override;
private:
- unsigned int _num_outputs;
+ unsigned int _num_outputs;
+ FullyConnectedLayerInfo _info;
};
} // namespace graph
} // namespace arm_compute