torch.fx.experimental.migrate_gradual_types.constraint_generator.masked_fill_inference_rule

torch.fx.experimental.migrate_gradual_types.constraint_generator.masked_fill_inference_rule(n, symbols, constraints, counter)

Similar to addition. For now we implement the constraints when the argument is a boolean tensor. There is also a case for when it is a condition. We will leave this out for now.