Searched refs:_DynamicLossScaleState (Results 1 – 1 of 1) sorted by relevance
215 class _DynamicLossScaleState(trackable.Trackable): class223 super(_DynamicLossScaleState, self).__init__()287 return (super(_DynamicLossScaleState, self)._checkpoint_dependencies +292 unconditional = super(_DynamicLossScaleState, self)._lookup_dependency(name)545 self._loss_scale = _DynamicLossScaleState(564 return isinstance(self._loss_scale, _DynamicLossScaleState)569 if isinstance(self._loss_scale, _DynamicLossScaleState):586 if isinstance(self._loss_scale, _DynamicLossScaleState):598 if isinstance(self._loss_scale, _DynamicLossScaleState):612 if isinstance(self._loss_scale, _DynamicLossScaleState):[all …]