pytorch

3ad3d8bd - [JIT] Add a pass for annotating graph with input types derived from sample inputs. (#57076)

[JIT] Add a pass for annotating graph with input types derived from sample inputs. (#57076) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57076 This pass is intended to be used in conjunction with shape propagation pass: first we use sample inputs to specify shape info for graph inputs and then we run shape-prop to infer shapes of intermediate values in the graph. Differential Revision: D28048290 Test Plan: Imported from OSS Reviewed By: astaff Pulled By: ZolotukhinM fbshipit-source-id: 778d772e873d59d77af9f669f45dc44b9ee5e443