Searched defs:memory_mask (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | transformer.py | 1560 memory_mask=None, argument
1648 …def _check_input(self, hidden_states, attention_mask, encoder_output, memory_mask, init_reset, bat… argument
2116 def construct(self, hidden_states, attention_mask, encoder_output=None, memory_mask=None, argument
2368 memory_mask=None, argument
|