Searched defs:q_trans (Results 1 – 1 of 1) sorted by relevance
1638 const AnfNodePtr &q_trans, const AnfNodePtr &k_trans, in CreateFAForSD15()1669 … const AnfNodePtr &q_trans, const AnfNodePtr &k_trans, in CreateFAWithPadAndPse()1943 auto q_trans = q_reshape->input(1)->cast<CNodePtr>(); in CreateFlashAttentionNodeForMsSD21() local2009 auto q_trans = q_reshape->input(1)->cast<CNodePtr>(); in CreateFlashAttentionNodeForVideoComposer() local2109 auto q_trans = q_reshape->input(kNumIndex1); in CreateFlashAttentionNodeForSD() local2291 auto q_trans = q_reshape->input(kNumIndex1); in CreateFlashAttentionNodeForSDWithoutCast() local2560 auto q_trans = q_reshape->input(kNumIndex1); in CreateFlashAttentionNodeForSDEinsum() local