original_docstring: PS embeddingLookup cache enable set.
docstring: PS embeddingLookup cache enable set.
docstring_tokens: PS embeddingLookup cache enable set .
code: def _set_voacb_cache_enable_for_ps(self, vocab_cache_size, embedding_size, vocab_size):
          """PS embeddingLookup cache enable set."""
          self.embedding_table.cache_enable = True
          self.embedding_table.is_param_ps = True
          _set_cache_enable(True)
          if self.sparse:
              self.forward_unique = True
          if _is_role_worker():
              _insert_hash_table_size(self.embedding_table.name, vocab_cache_size, embedding_size, vocab_size)
code_tokens: def _set_voacb_cache_enable_for_ps ( self , vocab_cache_size , embedding_size , vocab_size ) : self . embedding_table . cache_enable = True self . embedding_table . is_param_ps = True _set_cache_enable ( True ) if self . sparse : self . forward_unique = True if _is_role_worker ( ) : _insert_hash_table_size ( self . embedding_table . name , vocab_cache_size , embedding_size , vocab_size )
short_docstring: PS embeddingLookup cache enable set.
short_docstring_tokens: PS embeddingLookup cache enable set .
comment: """PS embeddingLookup cache enable set."""