Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Ddecoder_layer_fusion.cc405 std::shared_ptr<ops::DecoderLayer> DecoderLayerFusion::CreatePrim(const FuncGraphPtr &func_graph, c… in CreatePrim() function in mindspore::opt::DecoderLayerFusion
Dmulti_head_attention_fusion.cc894 std::shared_ptr<ops::Attention> MultiHeadAttentionFusion::CreatePrim(const EquivPtr &equiv, bool cr… in CreatePrim() function in mindspore::opt::MultiHeadAttentionFusion
Dencoder_layer_fusion.cc992 std::shared_ptr<ops::EncoderLayer> EncoderLayerFusion::CreatePrim(const FuncGraphPtr &func_graph, c… in CreatePrim() function in mindspore::opt::EncoderLayerFusion