Searched defs:fa_node (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | flash_attention_fusion.cc | 1658 …auto fa_node = CreatePromptFlashAttentionCnodeForBNSD(func_graph, node, q_pad_node, k_pad_node, v_… in CreateFAForSD15() local
1689 auto fa_node = in CreateFAWithPadAndPse() local
1794 …auto fa_node = CreatePromptFlashAttentionCnodeForBNSD(func_graph, node, q, k, v, nullptr, num_head… in CreateFlashAttentionNodeForMsSDXL() local
1906 CNodePtr fa_node = nullptr; in CreateFlashAttentionNodeForMsSDPseShift() local
1978 …auto fa_node = CreatePromptFlashAttentionCnodeForBNSD(func_graph, node, q_trans, k_trans, v_trans,… in CreateFlashAttentionNodeForMsSD21() local
2059 CNodePtr fa_node = nullptr; in CreateFlashAttentionNodeForVideoComposer() local
2149 CNodePtr fa_node = nullptr; in CreateFlashAttentionNodeForSD() local
2238 CNodePtr fa_node = nullptr; in CreateFlashAttentionNodeForSDPreMul() local
2341 CNodePtr fa_node = nullptr; in CreateFlashAttentionNodeForSDWithoutCast() local
2600 CNodePtr fa_node = nullptr; in CreateFlashAttentionNodeForSDEinsum() local
|