trunk/5e0379393cc64f15209ec20e20bc2eddc5ed2a42: Add autograd support to TokenSwitch dispatch and combine (#181314)

«dispatch and combine are adjoint operations: backward of dispatch calls combine, and backward of combine calls dispatch. Single dispatch/combine public API following nn.Module/torch.matmul conventions: With out=(out_tok...»