Searched refs:initial_loss (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | minimize_loss_test.py | 471 initial_loss = lambda: constant_op.constant(1e7) function
480 distribution.extended.call_for_each_replica(initial_loss))
482 "replica_loss_reduced": initial_loss(),
483 "cross_replica_loss_reduced": initial_loss(),
492 initial_loss(),
496 initial_loss(),
500 initial_loss(),
525 def _verify_loss_output(self, initial_loss, loss_output, reduced, argument
536 self.assertEqual(initial_loss.dtype, loss_tensor.dtype)
537 self.assertEqual(initial_loss.shape, loss_tensor.shape)
|