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 /examples/graph_deepspeech_v0_4_1.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 'examples/graph_deepspeech_v0_4_1.cpp')
-rw-r--r-- | examples/graph_deepspeech_v0_4_1.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/graph_deepspeech_v0_4_1.cpp b/examples/graph_deepspeech_v0_4_1.cpp index ed44ffbee2..cc65bf309d 100644 --- a/examples/graph_deepspeech_v0_4_1.cpp +++ b/examples/graph_deepspeech_v0_4_1.cpp @@ -289,7 +289,7 @@ private: tanh_ss.forward_tail(tanh_nid); // Add (third split) - NodeID add_nid = graph.graph().add_node<EltwiseLayerNode>(EltwiseOperation::Add); + NodeID add_nid = graph.graph().add_node<EltwiseLayerNode>(descriptors::EltwiseLayerDescriptor{ EltwiseOperation::Add }); graph.graph().add_connection(split_nid, 2, add_nid, 0); graph.graph().add_connection(add_y.tail_node(), 0, add_nid, 1); set_node_params(graph.graph(), add_nid, add_params); |