Searched defs:real_shift (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | flash_attention_score.cc | 45 …onst BaseTensorPtr &key, const BaseTensorPtr &value, const std::optional<BaseTensorPtr> &real_shif… in FlashAttentionScoreAscendCall()
100 const std::optional<BaseTensorPtr> &real_shift, const std::optional<BaseTensorPtr> &drop_mask, in FlashAttentionScoreAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_ops_flash_attention_score.py | 28 def flash_attention_score_forward_func(query, key, value, head_num, real_shift=None, drop_mask=None, argument
93 … input_layout='TND', real_shift=None, drop_mask=None, padding_mask=None, argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/auto_parallel/ |
| D | flash_attention_score.py | 84 …def construct(self, query, key, value, real_shift, attn_mask, actual_seq_qlen=None, actual_seq_kvl… argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_ring_attention.py | 117 def construct(self, query, key, value, real_shift, attn_mask): argument
|
| D | test_flash_attention_score_info.py | 159 …def construct(self, query, key, value, real_shift, attn_mask, actual_seq_qlen=None, actual_seq_kvl… argument
|
| D | test_model_flops_cal.py | 375 def construct(self, query, key, value, real_shift, attn_mask): argument
|
| /third_party/openssl/crypto/ |
| D | params.c | 22 static unsigned int real_shift(void) in real_shift() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | flash_attention_score_info.cc | 1136 … const AnfNodePtr &real_shift, const AnfNodePtr &drop_mask, in GetFlashAttentionScoreOpNode()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/manually_defined/ |
| D | ops_def.py | 1871 def flash_attention_score(query, key, value, head_num, real_shift=None, drop_mask=None, padding_mas… argument
|