ROIAlign.hpp

// Copyright (c) 2021 Graphcore Ltd. All rights reserved.
 
#ifndef popnn_ROIAlign_hpp
#define popnn_ROIAlign_hpp
#include <poplar/Program.hpp>
 
namespace popnn::experimental {
 
struct roiAlignParams {
  long unsigned int samplingRatio_, alignedHeight_, alignedWidth_;
  bool aligned_;
  float spatialScale_;
  roiAlignParams(long unsigned int samplingRatio, long unsigned int poolH,
                 long unsigned int poolW, bool aligned, float spatialScale)
      : samplingRatio_{samplingRatio}, alignedHeight_{poolH},
        alignedWidth_{poolW}, aligned_{aligned}, spatialScale_{spatialScale} {}
};
 
poplar::Tensor
roiAlignFwd(poplar::Graph &graph, poplar::program::Sequence &prog,
            poplar::Tensor &bottomData, poplar::Tensor &bottomRois,
            poplar::Tensor &bottomBatchIndex, const roiAlignParams &params,
            const poplar::DebugContext &debugContext = {});
 
poplar::Tensor
roiAlignInputGradient(poplar::Graph &graph, poplar::program::Sequence &prog,
                      poplar::Tensor &bottomData, poplar::Tensor &bottomRois,
                      poplar::Tensor &bottomBatchIndex,
                      poplar::Tensor &topDataGrad, const roiAlignParams &params,
                      const poplar::DebugContext &debugContext = {});
 
} // namespace popnn::experimental
 
#endif // popnn_ROIAlign_hpp