Since single dispatch for to_backend requires the first argument to be a valid class, we create the following dataclass spec to hold the dictionaries mapping the method name to the corresponding program, partitioner
| 656 | |
| 657 | @dataclass |
| 658 | class MethodProgramsPartitionerSpec: |
| 659 | """ |
| 660 | Since single dispatch for to_backend requires the first argument to be a |
| 661 | valid class, we create the following dataclass spec to hold the dictionaries |
| 662 | mapping the method name to the corresponding program, partitioner |
| 663 | """ |
| 664 | |
| 665 | method_to_edge_program: Mapping[str, ExportedProgram] |
| 666 | method_to_partitioner: Mapping[str, Partitioner] |
| 667 | |
| 668 | |
| 669 | @to_backend.register |
no outgoing calls