Searched defs:param_init (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 223 def __init__(self, vocab_size, embedding_size, param_init='normal', argument
476 def __init__(self, vocab_size, embedding_size, field_size, param_init='normal', target='CPU', argument
|
| D | thor_layer.py | 720 def __init__(self, vocab_size, embedding_size, param_init='normal', argument
|
| /third_party/mindspore/tests/st/auto_parallel/ |
| D | multifieldembeddinglookup_parallel.py | 179 param_init, target, slice_mode, sparse, operator, indices, field_ids): argument
203 param_init, target, slice_mode, sparse, operator, indices, field_ids): argument
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | transformer.py | 520 param_init='normal'): argument
|