Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/
Dgen_lite_ops.cc807 void FlashAttentionScoreGrad::set_input_layout(const int64_t &input_layout) { (void)thi… in set_input_layout() function in mindspore::ops::FlashAttentionScoreGrad
842 void FlashAttentionScore::set_input_layout(const int64_t &input_layout) { (void)this->A… in set_input_layout() function in mindspore::ops::FlashAttentionScore
2188 void FusedInferAttentionScore::set_input_layout(const int64_t &input_layout) { (void)th… in set_input_layout() function in mindspore::ops::FusedInferAttentionScore