Searched defs:use_one_hot_embeddings (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/tests/st/networks/models/bert/src/ |
| D | bert_model.py | 119 use_one_hot_embeddings=False, argument
173 use_one_hot_embeddings=False, argument
320 use_one_hot_embeddings=False): argument
423 use_one_hot_embeddings=False, argument
608 use_one_hot_embeddings=False, argument
679 use_one_hot_embeddings=False, argument
749 use_one_hot_embeddings=False, argument
845 use_one_hot_embeddings=False): argument
|
| D | bert_for_pre_training.py | 159 def __init__(self, config, is_training, use_one_hot_embeddings): argument
236 def __init__(self, config, is_training, use_one_hot_embeddings=False): argument
|
| /third_party/mindspore/tests/mindspore_test_framework/apps/ |
| D | test_bert_parts.py | 40 def __init__(self, config, is_training, use_one_hot_embeddings=True): argument
55 def __init__(self, config, is_training, use_one_hot_embeddings): argument
|
| D | bert_attention_submodules.py | 96 use_one_hot_embeddings=False, argument
287 use_one_hot_embeddings=False, argument
|
| /third_party/mindspore/mindspore/lite/examples/export_models/models/ |
| D | tinybert_train_export.py | 51 def __init__(self, student_config, is_training, use_one_hot_embeddings=False, argument
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | model.py | 58 def __init__(self,
315 use_one_hot_embeddings=False): argument
|