[FX] Add memory_format to shape_prop (#55815)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55815
Test Plan: Imported from OSS
Reviewed By: pbelevich, ansley
Differential Revision: D27716342
Pulled By: jamesr66a
fbshipit-source-id: f7c22dd77a4f48650700fc4c3c44b1c59196282e