Searched defs:k_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()1766 auto k_trans = matmul_1->input(2)->cast<CNodePtr>(); in CreateFlashAttentionNodeForMsSDXL() local1948 auto k_trans = k_reshape->input(1)->cast<CNodePtr>(); in CreateFlashAttentionNodeForMsSD21() local2016 auto k_trans = k_reshape->input(1)->cast<CNodePtr>(); in CreateFlashAttentionNodeForVideoComposer() local2111 auto k_trans = k_reshape->input(kNumIndex1); in CreateFlashAttentionNodeForSD() local2293 auto k_trans = k_reshape->input(kNumIndex1); in CreateFlashAttentionNodeForSDWithoutCast() local2562 auto k_trans = k_reshape->input(kNumIndex1); in CreateFlashAttentionNodeForSDEinsum() local