未验证 提交 39745d44 编写于 作者: D Double_V 提交者: GitHub

Roi align kl (#28128)

* add rois_num params for roi_align_xpu op, test=kunlun

* fix xpu roi_align_op warning,test=kunlun
上级 172e76a8
...@@ -62,7 +62,7 @@ class XPUROIAlignOpKernel : public framework::OpKernel<T> { ...@@ -62,7 +62,7 @@ class XPUROIAlignOpKernel : public framework::OpKernel<T> {
} else { } else {
auto _rois_lod = rois->lod().back(); auto _rois_lod = rois->lod().back();
rois_batch_size = _rois_lod.size() - 1; rois_batch_size = _rois_lod.size() - 1;
for (int n = 0; n < _rois_lod.size(); ++n) { for (int n = 0; n < static_cast<int>(_rois_lod.size()); ++n) {
rois_lod[n] = _rois_lod[n]; rois_lod[n] = _rois_lod[n];
} }
PADDLE_ENFORCE_EQ( PADDLE_ENFORCE_EQ(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册