Searched defs:fa_cnode (Results 1 – 3 of 3) sorted by relevance
52 … const CNodePtr &fa_cnode, in NewFlashAttentionCNodeWithAntiquantFusion()
255 auto fa_cnode = func_graph->NewCNode(fa_prim_c, {q_node, k_node, v_node, attn_mask_node}); in CreateFlashAttentionNodePart1() local
1745 auto fa_cnode = node->cast<CNodePtr>(); in PreProcessActualSeqLenInputForFlashAttentionScore() local1781 auto fa_cnode = node->cast<CNodePtr>(); in PostProcessActualSeqLenInputForFlashAttentionScore() local