From c30b668bf7b3e7f841ea8ef9295f43fc69519e15 Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Wed, 12 Sep 2018 17:44:08 +0100 Subject: COMPMID-1576: Add SliceLayer to graph API Change-Id: I42260cc529ca7721e221d3d0576b1668e09f8307 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/148255 Tested-by: bsgcomp Reviewed-by: Georgios Pinitas --- 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 b84abec9af..12c5a950af 100644 --- a/src/graph/GraphBuilder.cpp +++ b/src/graph/GraphBuilder.cpp @@ -509,6 +509,11 @@ NodeID GraphBuilder::add_softmax_node(Graph &g, NodeParams params, NodeIdxPair i return create_simple_single_input_output_node(g, params, input, beta); } +NodeID GraphBuilder::add_slice_node(Graph &g, NodeParams params, NodeIdxPair input, Coordinates &starts, Coordinates &ends) +{ + return create_simple_single_input_output_node(g, params, input, starts, ends); +} + NodeID GraphBuilder::add_split_node(Graph &g, NodeParams params, NodeIdxPair input, unsigned int num_splits, unsigned int axis) { return create_simple_single_input_output_node(g, params, input, num_splits, axis); -- cgit v1.2.1