Rate this Page

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:

tuple[list[Constraint], int]

Returns: Equality and consistency constraints on dimensions