Searched refs:_create_config_proto (Results 1 – 3 of 3) sorted by relevance
378 self._config_proto = self._create_config_proto()446 def _create_config_proto( member in TPUEmbedding
1375 self._config_proto = self._create_config_proto()1445 def _create_config_proto(self): member in TPUEmbedding
1298 return mid_level_api._create_config_proto()