Holds named data for serialization. Note: a DataEntry contains the index into `buffers`, the alignment and a tensor layout, if applicable. Attributes: buffers: A list of unique buffer entries. pte_data: Contains data that is stored inside the PTE file. A mapping from
| 35 | |
| 36 | @dataclass |
| 37 | class NamedDataStoreOutput: |
| 38 | """ |
| 39 | Holds named data for serialization. Note: a DataEntry contains the index into |
| 40 | `buffers`, the alignment and a tensor layout, if applicable. |
| 41 | |
| 42 | Attributes: |
| 43 | buffers: A list of unique buffer entries. |
| 44 | pte_data: Contains data that is stored inside the PTE file. A mapping from |
| 45 | {key: DataEntry}. |
| 46 | external_data: Contains data that is stored external to the PTE. A mapping |
| 47 | from {filename: {key: DataEntry}}. |
| 48 | """ |
| 49 | |
| 50 | buffers: List[bytes] |
| 51 | pte_data: Dict[str, DataEntry] |
| 52 | external_data: Dict[str, Dict[str, DataEntry]] |
| 53 | |
| 54 | |
| 55 | class NamedDataStore: |
no outgoing calls