Searched refs:keras_loss_scale_module (Results 1 – 2 of 2) sorted by relevance
23 from tensorflow.python.keras.mixed_precision import loss_scale as keras_loss_scale_module unknown371 self._loss_scale = keras_loss_scale_module.get(loss_scale)393 config['loss_scale'] = keras_loss_scale_module.serialize(self.loss_scale)400 config['loss_scale'] = keras_loss_scale_module.deserialize(
30 from tensorflow.python.keras.mixed_precision import loss_scale as keras_loss_scale_module unknown691 loss_scale = keras_loss_scale_module.deserialize(960 loss_scale = keras_loss_scale_module.deserialize(loss_scale)1024 config['loss_scale'] = keras_loss_scale_module.deserialize(