commit f8c4749a463106696ab99b6400b92a858a169a19
author Georgios Pinitas <georgios.pinitas@arm.com> Tue Feb 04 17:39:59 2020 +0000
committer Georgios Pinitas <georgios.pinitas@arm.com> Wed Feb 05 16:03:21 2020 +0000
tree 86d711d5bf25ef3b3a8288ba2aa0b25c2da61091
parent 55a687d5e2cf3434e4c2564ac73959dcc7163304

COMPMID-3083: Add graph support for PRelu

Change-Id: I48492f8c48b818b563bba1e0b48831ac5cdf1420
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2685
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>

arm_compute/graph/GraphBuilder.h

diff --git a/arm_compute/graph/GraphBuilder.h b/arm_compute/graph/GraphBuilder.h
index 33c222d..23541c4 100644
--- a/arm_compute/graph/GraphBuilder.h
+++ b/arm_compute/graph/GraphBuilder.h
@@ -366,6 +366,16 @@
      * @return Node ID of the created node, EmptyNodeID in case of error
      */
     static NodeID add_pooling_node(Graph &g, NodeParams params, NodeIdxPair input, PoolingLayerInfo pool_info);
+    /** Adds a prelu layer node to the graph
+     *
+     * @param[in] g      Graph to add the node to
+     * @param[in] params Common node parameters
+     * @param[in] input  Input to the PRelu node as a NodeID-Index pair
+     * @param[in] alpha  Alpha input to the PRelu node as a NodeID-Index pair
+     *
+     * @return Node ID of the created node, EmptyNodeID in case of error
+     */
+    static NodeID add_prelu_node(Graph &g, NodeParams params, NodeIdxPair input, NodeIdxPair alpha);
     /** Adds a print layer node to the graph
      *
      * @param[in] g           Graph to add the node to