aboutsummaryrefslogtreecommitdiff
path: root/src/graph/GraphBuilder.cpp
diff options
context:
space:
mode:
authorthecha01 <theo.charalambous@arm.com>2020-09-01 14:52:38 +0100
committerMichele Di Giorgio <michele.digiorgio@arm.com>2020-09-09 08:58:13 +0000
commit3603aff36444e01083b93e603e8b9b0a254d4911 (patch)
tree2210c3bd8a76654072a2bfa7b84c9151cafc2eee /src/graph/GraphBuilder.cpp
parent57f30a9309ff2e5e3b32731a785bf38b01d1fd69 (diff)
downloadComputeLibrary-3603aff36444e01083b93e603e8b9b0a254d4911.tar.gz
Add L2Normalize layer node to Graph API
Signed-off-by: thecha01 <theo.charalambous@arm.com> Change-Id: I5cd26a8829060563d63d8c53e5148631ee053eca Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3912 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/graph/GraphBuilder.cpp')
-rw-r--r--src/graph/GraphBuilder.cpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/graph/GraphBuilder.cpp b/src/graph/GraphBuilder.cpp
index ff85000e0c..33b7c178e5 100644
--- a/src/graph/GraphBuilder.cpp
+++ b/src/graph/GraphBuilder.cpp
@@ -552,6 +552,11 @@ NodeID GraphBuilder::add_generate_proposals_node(Graph &g, NodeParams params, No
return nid;
}
+NodeID GraphBuilder::add_l2_normalize_node(Graph &g, NodeParams params, NodeIdxPair input, int axis, float epsilon)
+{
+ return create_simple_single_input_output_node<L2NormalizeLayerNode>(g, params, input, axis, epsilon);
+}
+
NodeID GraphBuilder::add_normalization_node(Graph &g, NodeParams params, NodeIdxPair input, NormalizationLayerInfo norm_info)
{
return create_simple_single_input_output_node<NormalizationLayerNode>(g, params, input, norm_info);