Module shapes 

Functions

infer_graph_shapes

Run shape inference on a single graph.

infer_shapes

Run shape inference on all graphs in the program. Modifies node output_type in place.