def update()

in optimum/executorch/attentions/custom_kv_cache.py [0:0]

• 6 lines of code
• 1 McCabe index (conditional complexity)

    def update(
        self,
        key_states: torch.Tensor,
        value_states: torch.Tensor,
        layer_idx: int,
        cache_kwargs: Optional[Dict[str, Any]] = None,