Home
last modified time | relevance | path

Searched defs:input_layout_ (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dincre_flash_attention_info.h57 std::string input_layout_; variable
Dprompt_flash_attention_info.h58 std::string input_layout_; variable
Dfused_infer_attention_score_info.h73 int64_t input_layout_; variable
Dreshape_info.h115 TensorLayout input_layout_; variable
Dflash_attention_score_info.h131 int64_t input_layout_; // "BSH": 0; "BNSD": 1; variable