Searched defs:embedding_layer (Results 1 – 2 of 2) sorted by relevance
72 void EncoderLayer::set_embedding_layer(bool embedding_layer) { in set_embedding_layer()174 … bool layer_norm, bool use_past, bool query_layer, bool moe, bool embedding_layer) { in Init()
55 auto embedding_layer = GetValue<bool>(base_operator->GetAttr(ops::kEmbeddingLayer)); in IsSupport() local