Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dencoder_layer_fusion.cc995 auto encoder_layer_prim = std::make_shared<ops::EncoderLayer>(); in CreatePrim() local
1079 …auto encoder_layer_prim = CreatePrim(func_graph, equiv, ffn_hidden_size, expert_num, expert_offset… in CreateMaskedEncoderLayerFusionNode() local