Searched defs:CreateFlashAttentionNode (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ | ||
| D | flash_attention_fusion.cc | 208 CNodePtr FlashAttentionFusionV1::CreateFlashAttentionNode(const FuncGraphPtr &func_graph, const Anf… in CreateFlashAttentionNode() function in mindspore::opt::FlashAttentionFusionV1 |