| /external/tensorflow/tensorflow/python/keras/benchmarks/keras_examples_benchmarks/ |
| D | mnist_conv_custom_training_benchmark_test.py | 60 def compute_loss(self, targets, predictions, loss_fn, batch_size): argument
67 def train_step(self, inputs, model, loss_fn, optimizer, batch_size): argument
89 def distributed_train_step(self, batch_dataset, model, loss_fn, optimizer, argument
120 loss_fn, argument
176 loss_fn, argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | single_loss_example.py | 42 def loss_fn(ctx, x): function
68 def loss_fn(): function
105 def loss_fn(): function
|
| D | step_fn.py | 88 def __init__(self, dataset_fn, loss_fn, optimizer, distribution, argument
|
| D | collective_all_reduce_strategy_test.py | 127 def loss_fn(x): function
|
| D | parameter_server_strategy_test.py | 456 def loss_fn(x): function
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | test_example.py | 45 def loss_fn(): function
82 def loss_fn(): function
|
| D | keras_optimizer_v2_test.py | 64 def loss_fn(): function
|
| D | mirrored_strategy_test.py | 74 def loss_fn(ctx): function
|
| D | minimize_loss_test.py | 344 def loss_fn(): function
439 def loss_fn(): function
|
| D | distribute_strategy_test.py | 1667 def loss_fn(_, y_pred): member in TestRegularizerLoss
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | keras_test.py | 579 def loss_fn(y_true, y_pred): function
681 def loss_fn(y_true, y_pred): function
759 def loss_fn(y_true, y_pred): function
862 def loss_fn(y_true, y_pred): function
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_eager_v1.py | 36 def _eager_loss_fn(outputs, targets, loss_fn, output_name): argument
|
| D | training_v1.py | 2889 loss_fn, argument
2935 def loss_fn(self): member in _TrainingEndpoint
|
| D | training_utils_v1.py | 1134 def get_metric_function(metric, output_shape=None, loss_fn=None): argument
|
| D | training_test.py | 209 def loss_fn(labels, preds): function
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | learning_rate_schedule_test.py | 128 def loss_fn(): function
|
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | save_impl.py | 611 def _wrap_unconditional_loss(loss_fn, index): argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | losses_test.py | 212 def loss_fn(y_true, y_pred): function
1793 def loss_fn(y_true, y_pred): function
|
| D | losses.py | 1224 def _ragged_tensor_apply_loss(loss_fn, y_true, y_pred): argument
|