Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dself_define_shard_info.cc89 Status SelfDefineShardInfo::InferOutputTensorInfo() { in InferOutputTensorInfo() function in mindspore::parallel::SelfDefineShardInfo
Dbias_add_info.cc144 Status BiasAddInfo::InferOutputTensorInfo() { in InferOutputTensorInfo() function in mindspore::parallel::BiasAddInfo
Drms_norm_info.cc359 Status RmsNormInfo::InferOutputTensorInfo() { in InferOutputTensorInfo() function in mindspore::parallel::RmsNormInfo
Dlayer_norm_info.cc461 Status LayerNormInfo::InferOutputTensorInfo() { in InferOutputTensorInfo() function in mindspore::parallel::LayerNormInfo
Dtranspose_info.cc135 Status TransposeInfo::InferOutputTensorInfo() { in InferOutputTensorInfo() function in mindspore::parallel::TransposeInfo
Darithmetic_info.cc344 Status ArithmeticBase::InferOutputTensorInfo() { in InferOutputTensorInfo() function in mindspore::parallel::ArithmeticBase
Dactivation_info.cc95 Status ActivationBase::InferOutputTensorInfo() { in InferOutputTensorInfo() function in mindspore::parallel::ActivationBase
Doperator_info.h337 virtual Status InferOutputTensorInfo() { return SUCCESS; } in InferOutputTensorInfo() function
Dflash_attention_score_info.cc560 Status FlashAttentionScoreInfo::InferOutputTensorInfo() { in InferOutputTensorInfo() function in mindspore::parallel::FlashAttentionScoreInfo
Dmatmul_info.cc754 Status MatMul::InferOutputTensorInfo() { in InferOutputTensorInfo() function in mindspore::parallel::MatMul