Searched refs:_add_weight (Results 1 – 2 of 2) sorted by relevance
138 def _add_weight(self, name, initial_value, dtype=None): member in LossScale375 self._current_loss_scale = self._add_weight(381 self._num_good_steps = self._add_weight(
229 self._current_loss_scale = self._add_weight(236 self._counter = self._add_weight(239 def _add_weight(self, name, initial_value, dtype=None): member in _DynamicLossScaleState