- 30 Mar, 2021 1 commit
-
-
Nicolas Hug authored
* Add quantized version of nms * Added tests * Compute areas only once * remove calls to dequantize_val * fix return type for empty tensor * flake8 * remove use of scale as it gets cancelled out * simpler int convertion in tests * explicitly set ovr to double * add tests for more values of scale and zero_point * comment about underflow * remove unnecessary accessor * properly convert to float for division * Add comments about underflow * explicitely cast coordinates to float to allow vectorization * clang * clang again * hopefully OK now
-
- 15 Feb, 2021 1 commit
-
-
Vasilis Vryniotis authored
* Replace type T with accumulator. * Upcast tensors of box ops to avoid overflow in multiplications.
-
- 29 Jan, 2021 1 commit
-
-
Francisco Massa authored
Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.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.
-
- 14 Dec, 2020 1 commit
-
-
Vasilis Vryniotis authored
* Removing VISION_API from backward() methods and adding a ops.h * Fixing clang format.
-
- 11 Dec, 2020 2 commits
-
-
Vasilis Vryniotis authored
-
Vasilis Vryniotis authored
Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
-
- 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
-