(
result_vars: Mapping[Any, tuple[Variable, ...]], num_outputs: int
)
| 395 | |
| 396 | |
| 397 | def _unpack_dict_tuples( |
| 398 | result_vars: Mapping[Any, tuple[Variable, ...]], num_outputs: int |
| 399 | ) -> tuple[dict[Hashable, Variable], ...]: |
| 400 | out: tuple[dict[Hashable, Variable], ...] = tuple({} for _ in range(num_outputs)) |
| 401 | for name, values in result_vars.items(): |
| 402 | for value, results_dict in zip(values, out, strict=True): |
| 403 | results_dict[name] = value |
| 404 | return out |
| 405 | |
| 406 | |
| 407 | def _check_core_dims(signature, variable_args, name): |
no test coverage detected
searching dependent graphs…