pytorch
029289bd - [PyTorch] Migrate TensorAdvancedIndexing TI usage to borrowing where possible (#58276)

Commit
3 years ago
[PyTorch] Migrate TensorAdvancedIndexing TI usage to borrowing where possible (#58276) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58276 Borrowing is more efficient, and we can see in all these cases that the TensorIterator doesn't outlive the input & output Tensors. ghstack-source-id: 129002045 Test Plan: Existing CI Reviewed By: ngimel Differential Revision: D28428234 fbshipit-source-id: 9eada7725a070799b55e6683509e359505a2b80a
Author
Parents
Loading