torchvision/csrc/cpu/ROIAlign_cpu.cpp · 4480603831a6ad0bb97e3ffb5234f9d128e548a3 · OpenDAS / vision

Port roi_align to actually use dispatcher (#2366) · 44806038

Edward Z. Yang authored Jun 30, 2020



* Switch torchvision registrations to new operator registration API.

This is still registering everything as catchalls, so we're really just
moving deck chairs around, but payoff is coming soon.
Signed-off-by: Edward Z. Yang <ezyang@fb.com>

* Port roi_align to actually use dispatcher
Signed-off-by: Edward Z. Yang <ezyang@fb.com>

44806038

ROIAlign_cpu.cpp 14 KB

Replace ROIAlign_cpu.cpp