[nnc][aot_compiler] Memory formats args to aot_compiler (#72873)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/72873
Test Plan: Imported from OSS
Reviewed By: priyaramani
Differential Revision: D34250984
Pulled By: IvanKobzarev
fbshipit-source-id: e723ee64b024883eef78853e1b185b7040cafb09
(cherry picked from commit e9908df045acf33aa3cd0aec6784f15421236787)
