torch.fx.experimental.migrate_gradual_types.constraint_transformation.no_broadcast_dim_with_index

torch.fx.experimental.migrate_gradual_types.constraint_transformation.no_broadcast_dim_with_index(d1, d2, d3, d4, i)

Parameters:

• d1 (list[DVar]) – input 1

• d2 (list[DVar]) – input 2

• d3 (list[DVar]) – simulated broadcasting for input 1

• d4 (list[DVar]) – simulated broadcasting for input 2

• i (int) – the rank of the resulting tensor addition

Returns: Constraints for when no broadcasting occurs