Searched defs:dv_ (Results 1 – 1 of 1) sorted by relevance
815 std::optional<at::Tensor> &dv_, // batch_size x seqlen_k x num_heads_k x head_size in mha_bwd()1026 …std::optional<at::Tensor> &dv_, // total_k x num_heads_k x head_size, total_k := \sum_{i=0}^{b} … in mha_varlen_bwd()