Home
last modified time | relevance | path

Searched defs:new_sparse_mode (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dprompt_flash_attention_info.cc454 int64_t new_sparse_mode = is_attn_mask_compressed_ ? kSparseBand : sparse_mode_; in ReplaceNodeInputOrAttrs() local
Dflash_attention_score_info.cc1033 int64_t new_sparse_mode = is_attn_mask_compressed_ ? ops::kSparseBand : sparse_mode_; in ReplaceNodeInputOrAttrs() local
1139 int64_t new_sparse_mode = is_attn_mask_compressed_ ? ops::kSparseBand : sparse_mode_; in GetFlashAttentionScoreOpNode() local
Dfused_infer_attention_score_info.cc565 int64_t new_sparse_mode = is_attn_mask_compressed_ ? kSparseBand : sparse_mode_; in ReplaceNodeInputOrAttrs() local