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/backends/CL/CLFunctionsFactory.cpp

diff --git a/src/graph/backends/CL/CLFunctionsFactory.cpp b/src/graph/backends/CL/CLFunctionsFactory.cpp
index d627a39..f63aba9 100644
--- a/src/graph/backends/CL/CLFunctionsFactory.cpp
+++ b/src/graph/backends/CL/CLFunctionsFactory.cpp
@@ -119,6 +119,8 @@
             return detail::create_reshape_layer<CLReshapeLayer, CLTargetInfo>(*polymorphic_downcast<ReshapeLayerNode *>(node));
         case NodeType::ResizeLayer:
             return detail::create_resize_layer<CLScale, CLTargetInfo>(*polymorphic_downcast<ResizeLayerNode *>(node));
+        case NodeType::ROIAlignLayer:
+            return detail::create_roi_align_layer<CLROIAlignLayer, CLTargetInfo>(*polymorphic_downcast<ROIAlignLayerNode *>(node));
         case NodeType::SliceLayer:
             return detail::create_slice_layer<CLSlice, CLTargetInfo>(*polymorphic_downcast<SliceLayerNode *>(node));
         case NodeType::SoftmaxLayer: