Searched defs:batch_valid_length (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | transformer.py | 805 value_past=None, batch_valid_length=None): argument
886 value_past=None, batch_valid_length=None): argument
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 …(self, hidden_states, attention_mask, encoder_output, memory_mask, init_reset, batch_valid_length): 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
2370 batch_valid_length=None): argument
|