diff options
author | Sang-Hoon Park <sang-hoon.park@arm.com> | 2020-03-11 23:21:14 +0000 |
---|---|---|
committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2020-03-13 08:33:24 +0000 |
commit | 797b76b1aef38ea3be6f68ae2bf323048e9beff8 (patch) | |
tree | f78bb65ef0fe49bbb7d06225a4e7e4f2cc43cb45 /src/graph/GraphBuilder.cpp | |
parent | 470bc1eea65560d13001e60a7f7b22b12ec89bbc (diff) | |
download | ComputeLibrary-797b76b1aef38ea3be6f68ae2bf323048e9beff8.tar.gz |
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 <sang-hoon.park@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2876
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'src/graph/GraphBuilder.cpp')
-rw-r--r-- | src/graph/GraphBuilder.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
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<EltwiseLayerNode>(operation); + NodeID nid = g.add_node<EltwiseLayerNode>(descriptors::EltwiseLayerDescriptor{ operation }); g.add_connection(input0.node_id, input0.index, nid, 0); g.add_connection(input1.node_id, input1.index, nid, 1); |