Function _restore_named_data

exir/_serialize/_program.py:652–676

Moves named data from `segments` and `program` into the NamedDataStoreOutput class. Args: program: The Program holding named data references. Not modified. segments: The data containing the segments. Not modified.

(
    program: Program,
    segments: List[bytes],
)

def _restore_named_data(
    program: Program,
    segments: List[bytes],
) -> NamedDataStoreOutput:
    """Moves named data from `segments` and `program` into the
    NamedDataStoreOutput class.

    Args:
        program: The Program holding named data references. Not modified.
        segments: The data containing the segments. Not modified.
    """
    named_data_store = NamedDataStore()
    for entry in program.named_data:
        if entry.segment_index >= len(segments):
            raise ValueError(
                "Named data segment index "
                f"{entry.segment_index} >= num segments {len(segments)}"
            )
        named_data_store.add_named_data(
            key=entry.key,
            data=segments[entry.segment_index],
            alignment=1,  # Deserialization does not preserve alignment.
            tensor_layout=None,  # PTE file currently does not serialize this.
        )
    return named_data_store.get_named_data_store_output()


def _restore_segments(program: Program, segment_data: bytes) -> PTEFile: