tests/run_amp/test_multi_tensor_scale.py · 80a3f3cabd25cc1086cd9e5ffc7c8c3caab44e1a · OpenDAS / apex · GitLab

Find file Blame History Permalink

Allow multi-tensor unscale to handle FP16 output, so it can also be used for... · 80a3f3ca
Michael Carilli authored Feb 21, 2019
```
Allow multi-tensor unscale to handle FP16 output, so it can also be used for copy-scatter. Rename some options.
```
80a3f3ca

test_multi_tensor_scale.py 4.47 KB