Searched defs:embedding_table (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/distributed/embedding_cache/ |
| D | sparse_embedding_storage_test.cc | 49 std::unique_ptr<float[]> embedding_table = std::make_unique<float[]>(capacity * embedding_dim); in TEST_F() local
123 std::unique_ptr<float[]> embedding_table = std::make_unique<float[]>(capacity * embedding_dim); in TEST_F() local
|
| D | dense_embedding_storage_test.cc | 46 std::unique_ptr<float[]> embedding_table = std::make_unique<float[]>(capacity * embedding_dim); in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/distributed/persistent/ |
| D | persist_storage_test.cc | 53 PersistentData<float> embedding_table(data_ptr, embedding_shape); in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/layer/ |
| D | embedding.py | 104 def __init__(self, vocab_size, embedding_size, use_one_hot=False, embedding_table='normal', argument
|
| D | thor_layer.py | 584 def __init__(self, vocab_size, embedding_size, use_one_hot=False, embedding_table='normal', argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_dynamic_pangu_alpha_without_skip_redistribution.py | 889 def construct(self, state, embedding_table): argument
|
| D | test_dynamic_pangu_alpha.py | 885 def construct(self, state, embedding_table): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | encoder_layer_fusion.cc | 1053 …AnfNodePtr v_past, k_past, batch_valid_length, embedding_table, expert_ids, begin_expert_ids, expe… in CreateMaskedEncoderLayerFusionNode() local
|