From d64444ba197c2f95dcf4d205f50a196d5a29cdeb Mon Sep 17 00:00:00 2001 From: thecha01 Date: Mon, 7 Sep 2020 14:50:21 +0100 Subject: Add Reduction operation layer node to graph API Signed-off-by: thecha01 Change-Id: Ida819fb8c33790cc9da6d69eeb51e0599269197a Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3931 Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Comments-Addressed: Arm Jenkins --- src/graph/GraphBuilder.cpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/graph/GraphBuilder.cpp') diff --git a/src/graph/GraphBuilder.cpp b/src/graph/GraphBuilder.cpp index 33b7c178e5..2f74f065d5 100644 --- a/src/graph/GraphBuilder.cpp +++ b/src/graph/GraphBuilder.cpp @@ -642,6 +642,11 @@ NodeID GraphBuilder::add_quantization_node(Graph &g, NodeParams params, NodeIdxP return create_simple_single_input_output_node(g, params, input, out_quant_info); } +NodeID GraphBuilder::add_reduction_operation_node(Graph &g, NodeParams params, NodeIdxPair input, ReductionOperation op, int axis, bool keep_dims) +{ + return create_simple_single_input_output_node(g, params, input, op, axis, keep_dims); +} + NodeID GraphBuilder::add_reorg_node(Graph &g, NodeParams params, NodeIdxPair input, int stride) { return create_simple_single_input_output_node(g, params, input, stride); -- cgit v1.2.1