Searched defs:input_tensor_k_shape (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | flash_attention_fusion.cc | 183 auto input_tensor_k_shape = GetTensorShape(k_input_transpose, kNumIndex1); in CheckInputTensorShape() local
246 auto input_tensor_k_shape = GetTensorShape(k_input_transpose, kNumIndex1); in CreateFlashAttentionNode() local
360 auto input_tensor_k_shape = GetTensorShape(k_input_transpose, kNumIndex1); in CreateFlashAttentionNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | flash_attention_fusion.cc | 1485 auto input_tensor_k_shape = GetTensorShape(qk_matmul, kNumIndex2); in CreateFAForBNSDWithAttenMask() local
1557 auto input_tensor_k_shape = GetTensorShape(k_expend_dim, kNumIndex1); in CreateGQACNodeForBNSD() local
1774 auto input_tensor_k_shape = GetTensorShape(k_trans, kNumIndex1); in CreateFlashAttentionNodeForMsSDXL() local
1851 auto input_tensor_k_shape = GetTensorShape(k_reshape, kNumIndex1); in CreateFlashAttentionNodeForMsSDPseShift() local
1957 auto input_tensor_k_shape = GetTensorShape(k_reshape, 1); in CreateFlashAttentionNodeForMsSD21() local
2025 auto input_tensor_k_shape = GetTensorShape(k_reshape, 1); in CreateFlashAttentionNodeForVideoComposer() local
2117 auto input_tensor_k_shape = GetTensorShape(k_reshape, kNumIndex1); in CreateFlashAttentionNodeForSD() local
2193 auto input_tensor_k_shape = GetTensorShape(mul_k, kNumIndex1); in CreateFlashAttentionNodeForSDPreMul() local
2299 auto input_tensor_k_shape = GetTensorShape(k_reshape, kNumIndex1); in CreateFlashAttentionNodeForSDWithoutCast() local
2404 auto input_tensor_k_shape = GetTensorShape(matmul_1, kNumIndex2); in CreateFlashAttentionNodeForPanGu() local
[all …]
|