[vulkan][py] torch.utils.optimize_for_vulkan (#44903)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/44903
Test Plan: Imported from OSS
Reviewed By: kimishpatel
Differential Revision: D23766039
Pulled By: IvanKobzarev
fbshipit-source-id: dbdf484ee7d3a7719aab105efba51b92ebc51568
