Searched defs:output_trans (Results 1 – 1 of 1) sorted by relevance
2086 auto output_trans = reshape_o2->input(kNumIndex1)->cast<CNodePtr>(); in CreateFlashAttentionNodeForSD() local2174 auto output_trans = output_reshape->input(kNumIndex1)->cast<CNodePtr>(); in CreateFlashAttentionNodeForSDPreMul() local2270 auto output_trans = reshape_o2->input(kNumIndex1)->cast<CNodePtr>(); in CreateFlashAttentionNodeForSDWithoutCast() local2541 auto output_trans = reshape_o2->input(kNumIndex1)->cast<CNodePtr>(); in CreateFlashAttentionNodeForSDEinsum() local