def _extract_buffers_from_exir_program(
    exported_program,  # torch.export.ExportedProgram
) -> Dict[str, torch.Tensor]:
    """
    Given:
        exported_program: torch.export.ExportedProgram
    Return:
        buffers: dictionary mapping names to torch buffer tensors
    """
    buffers = {}
    for name, buffer in zip(
        exported_program.graph_signature.buffers, exported_program.buffers()
    ):
        if not isinstance(buffer, torch.Tensor):
            raise NotImplementedError(
                f"Only torch.Tensor buffer handled yet, but got {type(buffer)}"
            )
        buffers[name] = buffer
    return buffers

    Given:
        exported_program: torch.export.ExportedProgram
    Return:
        buffers: dictionary mapping names to torch buffer tensors
    
coremltools/converters/mil/frontend/torch/exir_utils.py