(
range_constraints: Dict[sympy.Symbol, ValueRanges]
)
| 378 | |
| 379 | |
| 380 | def serialize_range_constraints( |
| 381 | range_constraints: Dict[sympy.Symbol, ValueRanges] |
| 382 | ) -> Dict[str, RangeConstraint]: |
| 383 | return { |
| 384 | str(k): RangeConstraint( |
| 385 | _sympy_int_to_int(v.lower, "ceil"), # type: ignore[arg-type] |
| 386 | _sympy_int_to_int(v.upper, "floor"), # type: ignore[arg-type] |
| 387 | ) |
| 388 | for k, v in range_constraints.items() |
| 389 | } |
| 390 | |
| 391 | |
| 392 | def _is_single_tensor_return(target: torch._ops.OpOverload) -> bool: |
no test coverage detected