[GitHub] pengzhao-intel commented on a change in pull request #12738: Add Psroipooling CPU implementation

[GitBox <gi...@apache.org>]

pengzhao-intel commented on a change in pull request #12738: Add Psroipooling CPU implementation
URL: https://github.com/apache/incubator-mxnet/pull/12738#discussion_r223182861
 
 

 ##########
 File path: src/operator/contrib/psroi_pooling.cc
 ##########
 @@ -38,25 +38,191 @@ using std::floor;
 using std::ceil;
 
 namespace mshadow {
+
+template <typename DType>
+ inline void PSROIPoolForwardCPU(
+  const int count,
+  const DType* bottom_data,
+  const DType spatial_scale,
+  const int channels,
+  const int height, const int width,
+  const int pooled_height, const int pooled_width,
+  const DType* bottom_rois,
+  const int output_dim,
+  const int group_size,
+  DType* top_data) {
+  for (int index = 0; index < count; index++) {
 
 Review comment:
   Could you parallel this loop by OMP in case the computation is independent?

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services