COMPMID-1577: Add NormalizePlanarYUVLayer to graph API
Change-Id: Iadcf8205ff842330f7242f870705e9ea677ec419
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/148105
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
diff --git a/arm_compute/graph/GraphBuilder.h b/arm_compute/graph/GraphBuilder.h
index d750260..21a96a3 100644
--- a/arm_compute/graph/GraphBuilder.h
+++ b/arm_compute/graph/GraphBuilder.h
@@ -239,6 +239,18 @@
* @return Node ID of the created node, EmptyNodeID in case of error
*/
static NodeID add_normalization_node(Graph &g, NodeParams params, NodeIdxPair input, NormalizationLayerInfo norm_info);
+ /** Adds a normalize planar YUV 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 normalize planar YUV layer node as a NodeID-Index pair
+ * @param[in] mean_accessor Const Node ID that contains the mean values
+ * @param[in] std_accessor Const Node ID that contains the variance values
+ *
+ * @return Node ID of the created node, EmptyNodeID in case of error
+ */
+ static NodeID add_normalize_planar_yuv_node(Graph &g, NodeParams params, NodeIdxPair input,
+ ITensorAccessorUPtr mean_accessor = nullptr, ITensorAccessorUPtr std_accessor = nullptr);
/** Adds a permute layer node to the graph
*
* @param[in] g Graph to add the node to