Searched defs:query_tensor (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | transformer.py | 804 def construct(self, query_tensor, key_tensor, value_tensor, attention_mask, key_past=None, argument 885 def _check_inputs(self, query_tensor, key_tensor, value_tensor, attention_mask, key_past=None, argument 938 def _convert_to_2d_tensor(self, query_tensor, key_tensor, value_tensor, attention_mask): argument
|