| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_fused_infer_attention_score.py | 31 … deq_scale1, quant_scale1, deq_scale2, quant_scale2, quant_offset2, num_heads, scale_value=1.0, argument
52 dequant_scale1, quant_scale1, dequant_scale2, quant_scale2, quant_offset2, argument
80 … dequant_scale1, quant_scale1, dequant_scale2, quant_scale2, quant_offset2, antiquant_scale, argument
|
| D | test_prompt_flash_attention.py | 30 … deq_scale1, quant_scale1, deq_scale2, quant_scale2, quant_offset2, num_heads, scale_value=1.0, argument
|
| D | test_incre_flash_attention.py | 410 quant_scale1, argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_acl_ops/ |
| D | test_incre_flash_attention.py | 36 …self, query, key, value, attn_mask, actual_seq_lengths, padding_mask, dequant_scale1, quant_scale1, argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/auto_parallel/ |
| D | test_prompt_flash_attention_info.py | 138 deq_scale1, quant_scale1, deq_scale2, quant_scale2, quant_offset2): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_incre_flash_attention.py | 115 quant_scale1=None, argument
|
| D | test_prompt_flash_attention.py | 131 deq_scale1, quant_scale1, deq_scale2, quant_scale2, quant_offset2): argument
|
| D | test_fused_infer_attention_score.py | 170 … deq_scale1, quant_scale1, deq_scale2, quant_scale2, quant_offset2, antiquant_scale, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | nn_func.py | 7887 … deq_scale1, quant_scale1, deq_scale2, quant_scale2, quant_offset2, num_heads, argument
7962 …ttention(query, key, value, attn_mask, actual_seq_lengths, pse_shift, dequant_scale1, quant_scale1, argument
|