(self, device: str)
return self._get_device(device).loss_scaling_var
Get or create variable representing log2 of the current dynamic loss scaling factor.
Get or create variable representing log2 of the current dynamic loss scaling factor.
Get or create variable representing log2 of the current dynamic loss scaling factor .
def get_loss_scaling_var(self, device: str) -> Union[tf.Variable, None]:
        """Get or create variable representing log2 of the current dynamic loss scaling factor."""
        return self._get_device(device).loss_scaling_var
def get_loss_scaling_var ( self , device : str ) -> Union [ tf . Variable , None ] : return self . _get_device ( device ) . loss_scaling_var