Searched refs:_insert_hash_table_size (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/parallel/ |
| D | _ps_context.py | 201 def _insert_hash_table_size(name, cache_vocab_size, embedding_size, vocab_size): function
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 27 from mindspore.parallel._ps_context import _insert_hash_table_size, _set_cache_enable, _set_rank_id
379 … _insert_hash_table_size(self.embedding_table.name, vocab_cache_size, embedding_size, vocab_size)
|
| D | thor_layer.py | 907 … _insert_hash_table_size(self.embedding_table.name, vocab_cache_size, embedding_size, vocab_size)
|