Searched refs:keras_loss_scale_module (Results 1 – 2 of 2) sorted by relevance
28 from tensorflow.python.keras.mixed_precision import loss_scale as keras_loss_scale_module unknown378 self._loss_scale = keras_loss_scale_module.get(loss_scale)400 config['loss_scale'] = keras_loss_scale_module.serialize(self.loss_scale)407 config['loss_scale'] = keras_loss_scale_module.deserialize(
33 from tensorflow.python.keras.mixed_precision import loss_scale as keras_loss_scale_module unknown773 loss_scale = keras_loss_scale_module.deserialize(1021 loss_scale = keras_loss_scale_module.deserialize(loss_scale)1085 config['loss_scale'] = keras_loss_scale_module.deserialize(