Searched defs:query_layer (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | encoder_layer.cc | 70 void EncoderLayer::set_query_layer(bool query_layer) { (void)this->AddAttr(kQueryLayer, api::MakeVa… in set_query_layer()
174 … bool layer_norm, bool use_past, bool query_layer, bool moe, bool embedding_layer) { in Init()
|
| /third_party/mindspore/mindspore-src/source/tests/mindspore_test_framework/apps/ |
| D | bert_attention_submodules.py | 128 def construct(self, input_tensor, query_layer): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | encoder_tensorrt.cc | 53 auto query_layer = GetValue<bool>(base_operator->GetAttr(ops::kQueryLayer)); in IsSupport() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | encoder_layer_fusion.cc | 431 … bool query_layer = false, bool multi_batch = false, in DefinePatternEncoderSigma()
494 … bool is_layer_norm = false, bool query_layer = false, in DefinePatternEncoderAlpha()
|