vae example: provide an additional --xla flag to be able to run this test on TPU if available - #1416

vae example: provide an additional --xla flag to be able to run this test on TPU if available#1416

fedebongio wants to merge 2 commits into

pytorch:mainpytorch/examples:mainfrom

fedebongio:vae-xla-supportfedebongio/examples:vae-xla-supportCopy head branch name to clipboard