Searched defs:dk_expanded (Results 1 – 1 of 1) sorted by relevance
948 at::Tensor dk_expanded, dv_expanded; in mha_bwd() local1177 at::Tensor dk_expanded, dv_expanded; in mha_varlen_bwd() local