Home
last modified time | relevance | path

Searched defs:s1_split_num (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dflash_attention_score_info.h50 int64_t s1_split_num() { return s1_split_num_; } in s1_split_num() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_generate_strategy.cc296 int64_t s1_split_num = basic_stra[s_split_idx]; in PrepareFlashAttentionScore() local