Revert D31732421: [JIT][Easy] Shape cleanups
Test Plan: revert-hammer
Differential Revision:
D31732421 (https://github.com/pytorch/pytorch/commit/16d0896b690600aebccf8e545a4f63e226c49922)
Original commit changeset: e934507d1795
fbshipit-source-id: 6b34815c556de64ee5c7ef8d41e4cb434ccd7098
