| /third_party/mindspore/tests/st/auto_parallel/ |
| D | multifieldembeddinglookup_parallel.py | 179 param_init, target, slice_mode, sparse, operator, indices, field_ids): argument
183 param_init=param_init, target=target, slice_mode=slice_mode,
203 param_init, target, slice_mode, sparse, operator, indices, field_ids): argument
207 self.param_init = param_init
268 … field_size=self.field_size, param_init=self.param_init, target=self.target,
276 … field_size=self.field_size, param_init=self.param_init, target=self.target,
284 … field_size=self.field_size, param_init=self.param_init, target=self.target,
288 … field_size=self.field_size, param_init=self.param_init, target=self.target,
302 …fact = ParallelMultiHotFactory(vocab_size=32, embedding_size=64, field_size=64, param_init='one', …
|
| /third_party/skia/resources/sksl/runtime_errors/ |
| D | LoopInitializerErrors.rts | 14 void param_init(int p) { for (int i = p; i < 1; ++i) {} }
|
| /third_party/mindspore/tests/st/dynamic_shape/ |
| D | test_dynamic_shape_embedding.py | 32 param_init="ones", target=target)
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 223 def __init__(self, vocab_size, embedding_size, param_init='normal', argument
248 … self.embedding_table = Parameter(initializer(param_init, [self.vocab_size, self.embedding_size]),
476 def __init__(self, vocab_size, embedding_size, field_size, param_init='normal', target='CPU', argument
479 … super(MultiFieldEmbeddingLookup, self).__init__(vocab_size, embedding_size, param_init, target,
|
| D | thor_layer.py | 720 def __init__(self, vocab_size, embedding_size, param_init='normal', argument
747 … self.embedding_table = Parameter(initializer(param_init, [self.vocab_size, self.embedding_size],
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_sparse_feature_bprop.py | 84 self.embeddinglookup = nn.EmbeddingLookup(64, 64, param_init='ones')
|
| /third_party/mindspore/tests/st/ps/part_ps/ |
| D | test_ps_embedding_heterogeneous_conv2d_adam.py | 61 param_init='normal', target=target, sparse=sparse)
|
| /third_party/mindspore/tests/ut/python/ir/ |
| D | test_row_tensor.py | 441 …self.embedding_lookup = nn.EmbeddingLookup(vocab_size, embedding_size, param_init='ones', target=t…
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | transformer.py | 520 param_init='normal'): argument
525 … self.embedding_table = Parameter(initializer(param_init, [self.vocab_size, self.embedding_size]),
|