Searched refs:loss_output (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | minimize_loss_test.py | 493 loss_output=ctx.last_step_outputs["replica_loss_reduced"],
497 loss_output=ctx.last_step_outputs["cross_replica_loss_reduced"],
501 loss_output=ctx.last_step_outputs["cross_replica_loss_not_reduced"],
525 def _verify_loss_output(self, initial_loss, loss_output, reduced, argument
528 self.assertLen(distribution.experimental_local_results(loss_output),
530 loss_tensor = distribution.reduce(reduce_util.ReduceOp.MEAN, loss_output,
533 unwrapped_output = distribution.experimental_local_results(loss_output)
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | meta_graph.proto | 283 // key: "loss_output"
306 // key: "loss_output"
|