-
Notifications
You must be signed in to change notification settings - Fork 520
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add masked_index_select and refactor masked_index_put #2910
Conversation
✅ Deploy Preview for pytorch-fbgemm-docs ready!
To edit notification comments on pull requests, go to your Netlify site configuration. |
This pull request was exported from Phabricator. Differential Revision: D60362812 |
Differential Revision: D60348604
Differential Revision: D60356701
This pull request was exported from Phabricator. Differential Revision: D60362812 |
Summary: Pull Request resolved: pytorch#2910 This diff adds `torch.ops.fbgemm.masked_index_select` (will be used in subsequent diffs) and refactors `torch.ops.fbgemm.masked_index_put`. - Add unit tests - Add docstrings - Add contiguous tensor checks Differential Revision: D60362812
This pull request was exported from Phabricator. Differential Revision: D60362812 |
Summary: Pull Request resolved: pytorch#2910 This diff adds `torch.ops.fbgemm.masked_index_select` (will be used in subsequent diffs) and refactors `torch.ops.fbgemm.masked_index_put`. - Add unit tests - Add docstrings - Add contiguous tensor checks Differential Revision: D60362812
Summary: Pull Request resolved: pytorch#2910 This diff adds `torch.ops.fbgemm.masked_index_select` (will be used in subsequent diffs) and refactors `torch.ops.fbgemm.masked_index_put`. - Add unit tests - Add docstrings - Add contiguous tensor checks Reviewed By: q10 Differential Revision: D60362812
This pull request was exported from Phabricator. Differential Revision: D60362812 |
This pull request has been merged in 1fa5a68. |
Summary:
This diff adds
torch.ops.fbgemm.masked_index_select
(will be used insubsequent diffs) and refactors
torch.ops.fbgemm.masked_index_put
.Differential Revision: D60362812