pytorch/torch/testing/_internal/opinfo/definitions
Isalia20 80eff6e720 [MPS] fix triangular for >3D tensors (#144545)
Old implementation leads to incorrect output due to not handling the other batch sizes other than 3D tensors(B, M, N)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/144545
Approved by: https://github.com/malfet
2025-01-14 08:25:01 +00:00
..
__init__.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
_masked.py Update torch.masked.mean to upcast dtype for bool tensors (#139999) 2025-01-08 10:35:19 +00:00
fft.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
linalg.py [MPS] fix triangular for >3D tensors (#144545) 2025-01-14 08:25:01 +00:00
nested.py Migrate from Tuple -> tuple in torch/testing (#144256) 2025-01-10 06:37:55 +00:00
signal.py [BE][Opinfo] Delete redundant dtypesIfCUDA (#144512) 2025-01-10 15:15:38 +00:00
sparse.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
special.py [BE][Opinfo] Delete redundant dtypesIfCUDA (#144512) 2025-01-10 15:15:38 +00:00