torchrec
torchrec copied to clipboard
copy _permute_tensor_by_segments to avoid packaging import error
Summary:
packaging issue importing _permute_tensor_by_segments - jagged_tensor.py
refactor once jagged_tensor has the changes
Differential Revision: D59014188
This pull request was exported from Phabricator. Differential Revision: D59014188