• D
    Support prroi_pool_op with Tensor and LoDTensor rois (#20649) · 6ea38091
    Double_V 提交于
    1. Add a new input named batch_roi_nums for prroi_pool_op. batch_roi_nums includes the number of roi for each image in batch when rois is Tensor. This information is saved in rois's lod when rois is LoDTensor.
    2. add grad check to prroi_pool_op and solve unnormal X grad diff in CPU.
    6ea38091
prroi_pool_op.cu 14.9 KB