(var)
| 83 | |
| 84 | |
| 85 | def _parse_var_as_value(var): |
| 86 | assert isinstance( |
| 87 | var.bound_data, (np.ndarray, int) |
| 88 | ), "cannot parse a non-const var as value" |
| 89 | if tuple(var.bound_data.shape) == (1,): |
| 90 | return int(var.bound_data) |
| 91 | else: |
| 92 | return var.bound_data |
| 93 | |
| 94 | |
| 95 | def _can_broadcast_to(src, dst, broadcast_dims=None): |