torch.fx.experimental.migrate_gradual_types.constraint_generator.linear_inference_rule#

torch.fx.experimental.migrate_gradual_types.constraint_generator.linear_inference_rule(n, module_instance, symbols, constraints, counter)[source]#

Input and output sizes should be the same except for the last dimension If the input is Dyn, then so should the output