Home
last modified time | relevance | path

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/
Dflash_attention_score.cc45 …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/
Dtest_ops_flash_attention_score.py28 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/
Dflash_attention_score.py84 …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/
Dtest_ring_attention.py117 def construct(self, query, key, value, real_shift, attn_mask): argument
Dtest_flash_attention_score_info.py159 …def construct(self, query, key, value, real_shift, attn_mask, actual_seq_qlen=None, actual_seq_kvl… argument
Dtest_model_flops_cal.py375 def construct(self, query, key, value, real_shift, attn_mask): argument
/third_party/openssl/crypto/
Dparams.c22 static unsigned int real_shift(void) in real_shift() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dflash_attention_score_info.cc1136const AnfNodePtr &real_shift, const AnfNodePtr &drop_mask, in GetFlashAttentionScoreOpNode()
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/manually_defined/
Dops_def.py1871 def flash_attention_score(query, key, value, head_num, real_shift=None, drop_mask=None, padding_mas… argument