fix dynamic_shapes spec for moco (#148772)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/148333
X-link: https://github.com/pytorch/pytorch/pull/148772
Approved by: https://github.com/yushangdi, https://github.com/desertfire
Differential Revision: D71412041
