Home
last modified time | relevance | path

Searched defs:param_init (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/layer/
Dembedding.py254 def __init__(self, vocab_size, embedding_size, param_init='normal', argument
393 … def _set_voacb_cache_enable_for_ps(self, vocab_cache_size, embedding_size, vocab_size, param_init, argument
412 def _slice_pserver_embeddings(self, param_init, dtype=mstype.float32): argument
610 def __init__(self, vocab_size, embedding_size, field_size, param_init='normal', target='CPU', argument
Dthor_layer.py736 def __init__(self, vocab_size, embedding_size, param_init='normal', argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/
Dtest_autoparallel_rec_shared_param_strmodif.py52 param_init='normal', argument
Dtest_auto_parallel_frontend_scheduling_with_pangu_alpha.py460 def __init__(self, vocab_size, embedding_size, param_init='normal'): argument
Dtest_auto_parallel_pangu_alpha.py461 def __init__(self, vocab_size, embedding_size, param_init='normal'): argument
/third_party/mindspore/mindspore-src/source/tests/st/map_parameter/network/src/
Dmodel.py43 …def __init__(self, embedding_size, key_dtype=ms.int32, param_init='normal', max_norm=None, sparse=… argument
/third_party/mindspore/mindspore-src/source/tests/st/auto_parallel/
Dmultifieldembeddinglookup_parallel.py179 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-src/source/mindspore/python/mindspore/parallel/_transformer/
Dtransformer.py732 param_init='normal'): argument
/third_party/ffmpeg/libavfilter/
Dvf_convolution.c704 static int param_init(AVFilterContext *ctx) in param_init() function