Searched refs:_multiplier (Results 1 – 5 of 5) sorted by relevance
314 self._multiplier = float(multiplier)335 return self._multiplier365 new_loss_scale = self._current_loss_scale * self._multiplier379 self._current_loss_scale / self._multiplier, 1)
332 int _multiplier = producer.getMultiplier(magnitude + 1); in chooseMultiplierAndApply() local333 if (multiplier == _multiplier) { in chooseMultiplierAndApply()339 input.adjustMagnitude(_multiplier - multiplier); in chooseMultiplierAndApply()341 return _multiplier; in chooseMultiplierAndApply()
602 self._multiplier = linear_operator_util.convert_nonref_to_tensor(606 if not self._multiplier.dtype.is_complex:618 dtype=self._multiplier.dtype.base_dtype,740 math_ops.equal(self._multiplier, 0.),747 return self._multiplier
525 int _multiplier = producer.getMultiplier(magnitude + 1); in chooseMultiplierAndApply() local526 if (multiplier == _multiplier) { in chooseMultiplierAndApply()532 input.adjustMagnitude(_multiplier - multiplier); in chooseMultiplierAndApply()534 return _multiplier; in chooseMultiplierAndApply()
537 int _multiplier = producer.getMultiplier(magnitude + 1); in chooseMultiplierAndApply() local538 if (multiplier == _multiplier) { in chooseMultiplierAndApply()544 input.adjustMagnitude(_multiplier - multiplier); in chooseMultiplierAndApply()546 return _multiplier; in chooseMultiplierAndApply()