[Shape Inference] Infer shape info for second input of elementwise ops (#36365)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/36365
Test Plan: unit test
Reviewed By: yinghai, ipiszy
Differential Revision: D20959518
fbshipit-source-id: bafbe4f87534398003a49ff5bd8f398b3a0f1473
