Searched defs:seqused_k (Results 1 – 3 of 3) sorted by relevance
80 int * __restrict__ seqused_k; member
66 void *seqused_k, in set_params_fprop()549 …std::optional<at::Tensor> &seqused_k, // b. If given, only this many elements of each batch elemen… in mha_varlen_fwd()
865 std::optional<Tensor> seqused_k = _seqused_k; in _flash_attention_forward() local