Searched refs:_set_cache_enable (Results 1 – 3 of 3) sorted by relevance
27 from mindspore.parallel._ps_context import _insert_hash_table_size, _set_cache_enable, _set_rank_id308 self._set_cache_enable()315 def _set_cache_enable(self): member in EmbeddingLookup375 _set_cache_enable(True)
808 self._set_cache_enable()845 def _set_cache_enable(self): member in EmbeddingLookupThor903 _set_cache_enable(True)
221 def _set_cache_enable(cache_enable): function