- 10 May, 2021 1 commit
-
-
vfdev authored
* WIP Added antialias option to transforms.functional.resize * Updates according to the review * Excluded these C++ files for iOS build * Added support for mixed downsampling/upsampling * Fixed heap overflow caused by explicit loop unrolling * Applied PR review suggestions - used pytest parametrize instead unittest - cast to scalar_t ptr - removed interpolate aa files for ios/android keeping original cmake version
-
- 08 Apr, 2021 1 commit
-
-
Nicolas Hug authored
* WIP * clang * docs * extracted out common utils * Use better quantization function and pass tensors as parameters * proper dequantization * Some tests * Dequantization optimization, seems to gain a few ms * clang-format * again * more correct test. Had to remove optimization although it almost works * Also test aligned=True * remove useless part * more docs and comments * Put back optimization with more robust test * Added check for index upper bound * avoid possible overflow * Move common function into common.h * oops * scale=1,zero_point=0 makes more sense * Force batch size of 1 to prevent any indexingbug * format * format again * updated docstring * put back description comment for pre_calc_bilinear_interpolate * revert most changes to docstring as it's taken care of in another PR
-
- 31 Mar, 2021 1 commit
-
-
Nicolas Hug authored
* minor clean up * do same for ps_roialign Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
- 14 Jan, 2021 1 commit
-
-
Vasilis Vryniotis authored
-
- 04 Jan, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Adding TORCH_SELECTIVE_* macros on op registration. * Adding torchvision namespace.
-
- 10 Dec, 2020 1 commit
-
-
Vasilis Vryniotis authored
Summary: * Reduce unnecessary header inclusions in models and io. * Move autocast to separate folder and hide autograd implementation in an anonymous namespace. * Moving files in subfolders. Reviewed By: fmassa Differential Revision: D25461523 fbshipit-source-id: 756eeb6848aacaa474de4825ed4c1045d17e2cea
-