torch.fx.experimental.migrate_gradual_types.constraint_transformation.gen_consistency_constraints#
- torch.fx.experimental.migrate_gradual_types.constraint_transformation.gen_consistency_constraints(constraint, counter)[source]#
- Parameters:
constraint (BinConstraintT) – Consistency constraint on tensors
counter (int) – for variable tracking
- Return type:
Returns: Equality and consistency constraints on dimensions