Home
last modified time | relevance | path

Searched defs:prompt_flash_attention_cnode (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dflash_attention_fusion.cc137 CNodePtr prompt_flash_attention_cnode = nullptr; in CreatePromptFlashAttentionCnodeForBNSD() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dflash_attention_fusion.cc1337 CNodePtr prompt_flash_attention_cnode = nullptr; in CreatePromptFlashAttentionCnodeForBNSD() local
1392 CNodePtr prompt_flash_attention_cnode = nullptr; in CreatePromptFlashAttentionCnodeForBNSDWithPse() local
1449 CNodePtr prompt_flash_attention_cnode = nullptr; in CreatePromptFlashAttentionCnodeForBSH() local