Module shape_inference