commit 3f9d4d7c7242f50330dca4ff104a76bc88804903
author Manuel Bottini <manuel.bottini@arm.com> Fri Oct 19 14:04:42 2018 +0100
committer Georgios Pinitas <georgios.pinitas@arm.com> Fri Nov 16 13:02:50 2018 +0000
tree 28ec462d1d3d2bb7fa0a66de44b51798abfe96b0
parent 721c4cb0e1fbf1f151a79cd4c9d67e832f94f9bf

COMPMID-1637 - Add ROIAlign operator to the graph API

Change-Id: I8a9b1e16d90b9d99a6ff2a442347748432723b14

src/graph/GraphBuilder.cpp

diff --git a/src/graph/GraphBuilder.cpp b/src/graph/GraphBuilder.cpp
index 7870fb1..b4c5878 100644
--- a/src/graph/GraphBuilder.cpp
+++ b/src/graph/GraphBuilder.cpp
@@ -510,6 +510,20 @@
     return create_simple_single_input_output_node<ResizeLayerNode>(g, params, input, policy, width_scale, height_scale);
 }
 
+NodeID GraphBuilder::add_roi_align_node(Graph &g, NodeParams params, NodeIdxPair input, NodeIdxPair rois, ROIPoolingLayerInfo pool_info)
+{
+    CHECK_NODEIDX_PAIR(input, g);
+    CHECK_NODEIDX_PAIR(rois, g);
+
+    NodeID nid = g.add_node<ROIAlignLayerNode>(pool_info);
+
+    g.add_connection(input.node_id, input.index, nid, 0);
+    g.add_connection(rois.node_id, rois.index, nid, 1);
+
+    set_node_params(g, nid, params);
+    return nid;
+}
+
 NodeID GraphBuilder::add_scale_layer(Graph &g, const NodeParams &params, NodeIdxPair input, ITensorAccessorUPtr mul_accessor, ITensorAccessorUPtr add_accessor)
 {
     CHECK_NODEIDX_PAIR(input, g);