Searched refs:channels_unpooled (Results 1 – 4 of 4) sorted by relevance
153 int channels_unpooled = channels; in RoIAlignForwardKernel() local156 channels_unpooled = channels * pooled_height * pooled_width; in RoIAlignForwardKernel()159 bottom_data + (roi_batch_ind * channels_unpooled + c_unpooled) in RoIAlignForwardKernel()295 int channels_unpooled = channels; in RoIAlignBackwardKernel() local298 channels_unpooled = channels * pooled_height * pooled_width; in RoIAlignBackwardKernel()301 bottom_diff + (roi_batch_ind * channels_unpooled + c_unpooled) in RoIAlignBackwardKernel()
235 int channels_unpooled = channels; in ROIAlignForward() local238 channels_unpooled = channels * pooled_height * pooled_width; in ROIAlignForward()241 bottom_data + (roi_batch_ind * channels_unpooled + c_unpooled) in ROIAlignForward()378 int channels_unpooled = channels; in ROIAlignBackward() local381 channels_unpooled = channels * pooled_height * pooled_width; in ROIAlignBackward()384 bottom_diff + (roi_batch_ind * channels_unpooled + c_unpooled) in ROIAlignBackward()