Searched refs:loss_value (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | mnist_train.py | 149 loss_value = tf.reduce_mean(
152 grads = tape.gradient(loss_value, model.trainable_variables)
156 return accuracy, loss_value
164 loss_value = strategy.reduce(
166 return accuracy, loss_value
171 accuracy, loss_value = distributed_train_step(next(iterator))
174 tf.print(' Loss = %f' % loss_value)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 470 loss_value = loss()
476 loss_value = self._scale_loss(loss_value)
482 with ops.control_dependencies([loss_value]):
483 grads = tape.gradient(loss_value, var_list, grad_loss)
530 def _scale_loss(loss_value): argument
535 loss_value *= (1. / num_replicas)
537 return loss_value
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | losses_utils.py | 326 def scale_loss_for_distribution(loss_value): argument
331 loss_value *= (1. / num_replicas)
332 return loss_value
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | compile_utils.py | 205 loss_value = loss_obj(y_t, y_p, sample_weight=sw)
207 loss_metric_value = loss_value
222 loss_value *= loss_weight
227 loss_value = losses_utils.scale_loss_for_distribution(loss_value)
229 loss_values.append(loss_value)
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | gru_v2_test.py | 725 loss_value = history.history['loss'][0]
727 self.assertNotEqual(existing_loss, loss_value)
728 existing_loss = loss_value
|
| D | lstm_v2_test.py | 898 loss_value = history.history['loss'][0]
900 self.assertNotEqual(existing_loss, loss_value)
901 existing_loss = loss_value
|