Searched refs:extra_variables (Results 1 – 4 of 4) sorted by relevance
59 def gather_trainable_weights(trainable, sub_layers, extra_variables): argument78 v for v in extra_variables if v.trainable]82 def gather_non_trainable_weights(trainable, sub_layers, extra_variables): argument97 for v in extra_variables:
180 extra_variables=self._extra_variables)187 extra_variables=self._extra_variables)
244 def gather_trainable_weights(trainable, sub_layers, extra_variables): argument263 v for v in extra_variables if v.trainable]267 def gather_non_trainable_weights(trainable, sub_layers, extra_variables): argument282 for v in extra_variables:
690 extra_variables=self._trainable_weights)697 extra_variables=self._non_trainable_weights + self._trainable_weights)