torch.fx.experimental.unify_refinements.infer_symbolic_types

torch.fx.experimental.unify_refinements.infer_symbolic_types(traced)

Calls our symbolic inferencer twice. This is useful when one pass is not enough to infer all the information such as the case for broadcasting.