[JIT] Shape inference improvement (#35051)
Summary:
Support `aten::div` in `PropagateCompleteShapeOnNode`.
complete shape propagation on `aten::div` is disabled, because shape inference
relies on running node to propagate shape. For `aten::div` we run into
deviding-by-zero problem.
However, shape propagation for pointwise operatoins should be identical. We
would be able to swap the operation for `aten::div` with `aten::mul`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35051
Differential Revision: D20921359
Pulled By: eellison
fbshipit-source-id: 344371f34724a1b6bb2f853ebb4cef80423a4f9f