[jit] support tracing tensor __setitem__ with dynamic shape (#45828)
Summary:
fix (partially) https://github.com/pytorch/pytorch/issues/43548
Pull Request resolved: https://github.com/pytorch/pytorch/pull/45828
Test Plan: buck test mode/dev-nosan //caffe2/test:jit -- 'test_trace_slice' --jobs 1
Reviewed By: bdhirsh
Differential Revision: D24106641
Pulled By: ppwwyyxx
fbshipit-source-id: 8036c9819c9816e040796dac8f9c98bd33ce80a8