torch.fx.experimental.graph_gradual_typechecker.linear_inference_rule#

torch.fx.experimental.graph_gradual_typechecker.linear_inference_rule(n, module_instance)[source]#

Applies the shape information to the input then gets the greatest upper bound of the resulting type and the existing type