Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dflash_attention_fusion.cc2035 auto mul_const_input = mul->input(kNumIndex2); in CreateFlashAttentionNodeForVideoComposer() local
2126 auto mul_const_input = mul->input(kNumIndex2); in CreateFlashAttentionNodeForSD() local
2321 auto mul_const_input = mul->input(kNumIndex2); in CreateFlashAttentionNodeForSDWithoutCast() local
2577 auto mul_const_input = mul->input(kNumIndex2); in CreateFlashAttentionNodeForSDEinsum() local