def forward(self, *args):
        """Performs Multi Level RoiAlign on the input."""
        assert len(args) >= 2
        x = args[0]
        rois = args[1]
        return self.op(x, rois)
Performs Multi Level RoiAlign on the input.
python/aitemplate/frontend/nn/roi_ops.py