Function _layer_is_empty

KVCOMM/llm/kvcomm_engine.py:123–130

(tensor: Optional[torch.Tensor])

def _layer_is_empty(tensor: Optional[torch.Tensor]) -> bool:
    if tensor is None:
        return True
    if isinstance(tensor, list):
        return len(tensor) == 0
    if isinstance(tensor, torch.Tensor):
        return tensor.numel() == 0 or tensor.shape[-2] == 0
    return False


def _layer_length(tensor: Optional[torch.Tensor]) -> int: