| /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 | 413 quant_offset2, argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_acl_ops/ |
| D | test_incre_flash_attention.py | 37 dequant_scale2, quant_scale2, quant_offset2): 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 | 118 quant_offset2=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
7963 … dequant_scale2, quant_scale2, quant_offset2, antiquant_scale, antiquant_offset, block_table, argument
|