Use torch.accelerator API in VAE example #1338

msaroufim merged 2 commits into

pytorch:mainpytorch/examples:mainfrom

jafraustro:eromomon/accel_vaejafraustro/examples_xpu:eromomon/accel_vaeCopy head branch name to clipboard