Searched refs:vocab_cache_size (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 225 max_norm=None, sparse=True, vocab_cache_size=0): argument
230 … self.vocab_cache_size = validator.check_non_negative_int(vocab_cache_size, 'vocab_cache_size')
233 self.cache_enable = self.vocab_cache_size > 0
260 self._set_voacb_cache_enable_for_ps(vocab_cache_size, embedding_size, vocab_size)
332 self.embedding_table.cache_shape = (self.vocab_cache_size, self.embedding_size)
338 if self.vocab_cache_size > 0:
359 self.vocab_cache_size = self.vocab_cache_size * rank_size
363 self.vocab_size = self.vocab_cache_size
371 def _set_voacb_cache_enable_for_ps(self, vocab_cache_size, embedding_size, vocab_size): argument
379 … _insert_hash_table_size(self.embedding_table.name, vocab_cache_size, embedding_size, vocab_size)
|
| D | thor_layer.py | 722 max_norm=None, sparse=True, vocab_cache_size=0): argument
726 …self.vocab_cache_size = Validator.check_non_negative_int(vocab_cache_size, 'vocab_cache_size', sel…
729 self.cache_enable = self.vocab_cache_size > 0
759 self._set_voacb_cache_enable_for_ps(vocab_cache_size, embedding_size, vocab_size)
862 self.embedding_table.cache_shape = (self.vocab_cache_size, self.embedding_size)
868 if self.vocab_cache_size > 0:
889 self.vocab_cache_size = self.vocab_cache_size * rank_size
893 self.vocab_size = self.vocab_cache_size
899 def _set_voacb_cache_enable_for_ps(self, vocab_cache_size, embedding_size, vocab_size): argument
907 … _insert_hash_table_size(self.embedding_table.name, vocab_cache_size, embedding_size, vocab_size)
|
| /third_party/mindspore/mindspore/train/train_thor/ |
| D | convert_utils.py | 89 vocab_cache_size=subcell.vocab_cache_size)
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | unique_info.cc | 130 auto slice_size = SizeToLong(ps::PsCacheManager::GetInstance().vocab_cache_size()); in ComputeReplaceGraph()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ |
| D | ps_cache_manager.h | 131 size_t vocab_cache_size() const { return vocab_cache_size_; } in vocab_cache_size() function
|