Searched refs:get_scaled_loss (Results 1 – 8 of 8) sorted by relevance
115 scaled_loss = opt.get_scaled_loss(loss)
622 def get_scaled_loss(self, loss): member in LossScaleOptimizer680 loss = self.get_scaled_loss(loss)692 loss = self.get_scaled_loss(loss)
142 self.assertEqual(10., self.evaluate(opt.get_scaled_loss(loss)))143 self.assertEqual(10., self.evaluate(opt.get_scaled_loss(lambda: loss)()))145 self.assertEqual(10., self.evaluate(opt.get_scaled_loss(loss)))146 self.assertEqual(10., self.evaluate(opt.get_scaled_loss(lambda: loss)()))
81 name: "get_scaled_loss"
82 name: "get_scaled_loss"
264 scaled_total_loss = model.optimizer.get_scaled_loss(total_loss)