Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dprompt_flash_attention_info.cc236 …int64_t new_next_tokens = (sparse_mode_ == kSparseDefaultMask || sparse_mode_ == kSparseBand) ? ne… in GetAttenionMaskAttrs() local
452 int64_t new_pre_tokens, new_next_tokens; in ReplaceNodeInputOrAttrs() local
Dfused_infer_attention_score_info.cc420 …int64_t new_next_tokens = (sparse_mode_ == kSparseDefaultMask || sparse_mode_ == kSparseBand) ? ne… in GetAttentionMaskAttrs() local
563 int64_t new_next_tokens = 0; in ReplaceNodeInputOrAttrs() local
Dflash_attention_score_info.cc952 int64_t new_next_tokens = in GetAttentionMaskAttrs() local
1029 int64_t new_pre_tokens, new_next_tokens; in ReplaceNodeInputOrAttrs() local
1140 int64_t new_pre_tokens, new_next_tokens; in GetFlashAttentionScoreOpNode() local