Add support for arbitrary inputs and outputs in AutoParallel#311

fmassa wants to merge 3 commits intomainmeta-pytorch/autoparallel:mainfrom

fmassa/support_arbitrary_inputsmeta-pytorch/autoparallel:fmassa/support_arbitrary_inputsCopy head branch name to clipboard