Searched defs:encoder_output (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/_transformer/ |
| D | transformer.py | 2106 encoder_output=None, argument
2200 …def _check_input(self, hidden_states, attention_mask, encoder_output, memory_mask, init_reset, bat… argument
2751 def construct(self, hidden_states, attention_mask, encoder_output=None, memory_mask=None, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | decoder_layer_fusion.cc | 439 auto encoder_output = utils::cast<AnfNodePtr>((*equiv)[encoder_output_]); in CreateMaskedDecoderLayerFusionNode() local
|