| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_fused_infer_attention_score.py | 30 …construct(self, query, key, value, attn_mask, actual_seq_lengths, actual_seq_lengths_kv, pse_shift, argument
51 def construct(self, query, key_i, value_i, attn_mask, actual_seq_lengths, pse_shift, argument
79 …def construct(self, query, key, value, pse_shift, attn_mask, actual_seq_lengths, actual_seq_length… argument
|
| D | test_prompt_flash_attention.py | 29 …construct(self, query, key, value, attn_mask, actual_seq_lengths, actual_seq_lengths_kv, pse_shift, argument
|
| D | test_incre_flash_attention.py | 408 pse_shift, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | flash_attention_score_grad.cc | 32 const std::optional<BaseTensorPtr> &pse_shift, const std::optional<BaseTensorPtr> &drop_mask, in FlashAttentionScoreGradAscendCall()
86 …const BaseTensorPtr &dy, const std::optional<BaseTensorPtr> &pse_shift, const std::optional<BaseTe… in FlashAttentionScoreGradAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/tests/st/auto_parallel/ |
| D | test_prompt_flash_attention_info.py | 137 …construct(self, query, key, value, attn_mask, actual_seq_lengths, actual_seq_lengths_kv, pse_shift, argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_prompt_flash_attention.py | 130 …construct(self, query, key, value, attn_mask, actual_seq_lengths, actual_seq_lengths_kv, pse_shift, argument
|
| D | test_fused_infer_attention_score.py | 169 …def construct(self, query, key, value, pse_shift, attn_mask, actual_seq_lengths, actual_seq_length… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | nn_func.py | 7886 …flash_attention(query, key, value, attn_mask, actual_seq_lengths, actual_seq_lengths_kv, pse_shift, argument
7962 def incre_flash_attention(query, key, value, attn_mask, actual_seq_lengths, pse_shift, dequant_scal… argument
|