pytorch/torch/nested
David Berard 460950d3aa [Nested Tensor] Support ragged_idx != 1 on aten::is_same_size, aten::_to_copy (#118442)
is_same_size is needed internally; `_to_copy` should be easy because it doesn't support new layouts.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/118442
Approved by: https://github.com/cpuhrsch
2024-01-30 01:32:51 +00:00
..
_internal [Nested Tensor] Support ragged_idx != 1 on aten::is_same_size, aten::_to_copy (#118442) 2024-01-30 01:32:51 +00:00
__init__.py Fix return type hint for list types (#118238) 2024-01-25 23:35:20 +00:00