pytorch.py
35.5 KB
-
[Torch] Upsampling op support and enable registering a user defined op conversion map (#4961) · 92a24278
* add custom conversion map * add roi align test using custom convert map * refactor test * add support for upsampling op and test on segmentation models * remove redundant no_grad * add upsampling test case * make the default custom map None, instead of empty dict * updated tests, remove packaging and drop PT 1.2 support * add better support for aten::to and tests * add a note on dilation in x86
masahi committed