Searched defs:init_reset (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | transformer.py | 1239 def construct(self, x, input_mask, init_reset=True, batch_valid_length=None): argument
1312 def _check_input(self, x, input_mask, init_reset, batch_valid_length): argument
1561 init_reset=True, batch_valid_length=None): argument
1648 …def _check_input(self, hidden_states, attention_mask, encoder_output, memory_mask, init_reset, bat… argument
1922 def construct(self, hidden_states, attention_mask, init_reset=True, batch_valid_length=None): argument
2117 init_reset=True, batch_valid_length=None): argument
2369 init_reset=True, argument
|