Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dprompt_flash_attention_info.cc198 int64_t PromptFlashAttentionInfo::GetSplitIdAndRank() { in GetSplitIdAndRank() function in mindspore::parallel::PromptFlashAttentionInfo
Dfused_infer_attention_score_info.cc346 int64_t FusedInferAttentionScoreInfo::GetSplitIdAndRank() { in GetSplitIdAndRank() function in mindspore::parallel::FusedInferAttentionScoreInfo
Dflash_attention_score_info.cc923 std::vector<int64_t> FlashAttentionScoreInfo::GetSplitIdAndRank() { in GetSplitIdAndRank() function in mindspore::parallel::FlashAttentionScoreInfo