Searched refs:_set_voacb_cache_enable_for_ps (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 260 self._set_voacb_cache_enable_for_ps(vocab_cache_size, embedding_size, vocab_size)
371 def _set_voacb_cache_enable_for_ps(self, vocab_cache_size, embedding_size, vocab_size): member in EmbeddingLookup
|
| D | thor_layer.py | 759 self._set_voacb_cache_enable_for_ps(vocab_cache_size, embedding_size, vocab_size)
899 def _set_voacb_cache_enable_for_ps(self, vocab_cache_size, embedding_size, vocab_size): member in EmbeddingLookupThor
|