GitLab

* update roipool to make in torch fx traceable

* update ps_roi_align, ps_roi_pool and roi_align to make them torch fx traceable

* add unittest

* Remove `.to(output_fx)` from test
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>