Commits on Oct 25, 2022

1. Add group_index_select_dim0 …

   Summary:
   `group_index_select_dim0` does `index_select_dim0` for each group of
   inputs in a single kernel.  The operator takes a list of inputs and a
   list of indices and returns a list of outputs.
   
   There are some limitations to group_index_select_dim0:
   - All inputs must have the same shape.
   - All indices must have the same shape.
   - Because we use variadic template for the autograd function, it
   supports up to 55 groups.
   
   Differential Revision: D40683435
   
   fbshipit-source-id: 5c4e6adced3fad43206f91b1a9131774d6ed71f5

   

   sryap authored and facebook-github-bot committed Oct 25, 2022
   Configuration menu

   • View commit details
   • Copy full SHA for b10c170
   • Browse repository at this point

   Copy the full SHA

   b10c170 View commit details

   Browse the repository at this point in the history