Searched defs:num_attention_heads (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/tests/mindspore_test_framework/apps/ |
| D | bert_attention_submodules.py | 39 num_attention_heads=1, argument
94 num_attention_heads=1, argument
230 num_attention_heads=1, argument
285 num_attention_heads=1, argument
|
| /third_party/mindspore/tests/st/networks/models/bert/src/ |
| D | bert_model.py | 68 num_attention_heads=12, argument
416 num_attention_heads=1, argument
606 num_attention_heads=12, argument
676 num_attention_heads=12, argument
746 num_attention_heads=12, argument
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | model.py | 65 num_attention_heads=12, argument
|