Searched defs:output_shape_node (Results 1 – 1 of 1) sorted by relevance
2045 auto output_shape_node = node->cast<CNodePtr>(); in CreateFlashAttentionNodeForVideoComposer() local2135 auto output_shape_node = node->cast<CNodePtr>(); in CreateFlashAttentionNodeForSD() local2331 auto output_shape_node = node->cast<CNodePtr>(); in CreateFlashAttentionNodeForSDWithoutCast() local2586 auto output_shape_node = node->cast<CNodePtr>(); in CreateFlashAttentionNodeForSDEinsum() local