Searched refs:_load_variables (Results 1 – 2 of 2) sorted by relevance
104 self.first_mid_level._load_variables()112 self.second_mid_level._load_variables()130 self.first_mid_level._load_variables()138 self.second_mid_level._load_variables()268 self.first_mid_level._load_variables()
399 self._load_variables()855 def _load_variables(self): member in TPUEmbedding890 return TPUEmbeddingSaveable(name, self._load_variables,