From 797b76b1aef38ea3be6f68ae2bf323048e9beff8 Mon Sep 17 00:00:00 2001 From: Sang-Hoon Park Date: Wed, 11 Mar 2020 23:21:14 +0000 Subject: COMPMID-3221: Add EltwiseLayerDescriptor A new descriptor struct for EltwiseLayerNode is added to have better extendability. Change-Id: I3d0a4b3cec1f2425f39157cee6b5c344336412a3 Signed-off-by: Sang-Hoon Park Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2876 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio --- src/graph/GraphBuilder.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/graph/GraphBuilder.cpp') diff --git a/src/graph/GraphBuilder.cpp b/src/graph/GraphBuilder.cpp index 97ade9362d..e429817d50 100644 --- a/src/graph/GraphBuilder.cpp +++ b/src/graph/GraphBuilder.cpp @@ -438,7 +438,7 @@ NodeID GraphBuilder::add_elementwise_node(Graph &g, NodeParams params, NodeIdxPa check_nodeidx_pair(input0, g); check_nodeidx_pair(input1, g); - NodeID nid = g.add_node(operation); + NodeID nid = g.add_node(descriptors::EltwiseLayerDescriptor{ operation }); g.add_connection(input0.node_id, input0.index, nid, 0); g.add_connection(input1.node_id, input1.index, nid, 1); -- cgit v1.2.1