Move prim::layout for lite jit (#53781)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/53781
needed for running noise suppression model in lite interpreter
Test Plan: run model
Reviewed By: linbinyu
Differential Revision: D26967227
fbshipit-source-id: 19677fc796f1fb4423ebb11b5ffd9df5870a39cf