Module shapes

Module shapes 

Source

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.