Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Drms_norm_info.cc330 Status RmsNormInfo::InferOutputLayout() { in InferOutputLayout() function in mindspore::parallel::RmsNormInfo
Dlayer_norm_info.cc429 Status LayerNormInfo::InferOutputLayout() { in InferOutputLayout() function in mindspore::parallel::LayerNormInfo
Dtranspose_info.cc296 TensorLayout TransposeInfo::InferOutputLayout() { in InferOutputLayout() function in mindspore::parallel::TransposeInfo
Darithmetic_info.cc395 TensorLayout ArithmeticBase::InferOutputLayout() { in InferOutputLayout() function in mindspore::parallel::ArithmeticBase
Dflash_attention_score_info.cc526 Status FlashAttentionScoreInfo::InferOutputLayout() { in InferOutputLayout() function in mindspore::parallel::FlashAttentionScoreInfo
Dmatmul_info.cc714 TensorLayout MatMul::InferOutputLayout() { in InferOutputLayout() function in mindspore::parallel::MatMul