Home
last modified time | relevance | path

Searched defs:loss_fn (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/contrib/distribute/python/
Dsingle_loss_example.py41 def loss_fn(ctx, x): function
75 def loss_fn(): function
112 def loss_fn(): function
Dkeras_optimizer_v2_test.py65 def loss_fn(): function
Dminimize_loss_test.py296 def loss_fn(): function
388 def loss_fn(): function
Dstep_fn.py88 def __init__(self, dataset_fn, loss_fn, optimizer, distribution, argument
Dcollective_all_reduce_strategy_test.py161 def loss_fn(x): function
Dkeras_test.py1239 def loss_fn(_, y_pred): member in TestRegularizerLoss
Dparameter_server_strategy_test.py499 def loss_fn(x): function
Dmirrored_strategy_multigpu_test.py1401 def loss_fn(ctx): function
/external/tensorflow/tensorflow/contrib/gan/python/losses/python/
Dtuple_losses_impl.py89 def _args_to_gan_model(loss_fn): argument
285 def stargan_generator_loss_wrapper(loss_fn): argument
310 def stargan_discriminator_loss_wrapper(loss_fn): argument
336 def stargan_gradient_penalty_wrapper(loss_fn): argument
Dtuple_losses_test.py72 def loss_fn(x): function
/external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
Dlatent_gan_estimator_impl.py93 def _get_latent_gan_model_fn(generator_fn, discriminator_fn, loss_fn, argument
138 def get_latent_gan_estimator(generator_fn, discriminator_fn, loss_fn, argument
Dstargan_estimator_impl.py101 loss_fn=None, argument
209 loss_fn, argument
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dhead.py277 loss_fn=None, argument
395 loss_fn=None): argument
449 def loss_only_head(loss_fn, head_name=None): argument
632 loss_fn, argument
686 loss_fn, argument
818 loss_fn=None, argument
999 loss_fn=None, argument
1334 loss_fn=None): argument
1465 def __init__(self, loss_fn, head_name=None): argument
1843 def _wrap_custom_loss_fn(loss_fn): argument
[all …]
/external/tensorflow/tensorflow/contrib/distribute/python/examples/
Dsimple_estimator_example.py45 def loss_fn(): function
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dtarget_column.py151 def __init__(self, loss_fn, num_label_columns, label_name, weight_column_name, argument
266 def __init__(self, loss_fn, label_name, weight_column_name, label_dimension): argument
294 def __init__(self, loss_fn, n_classes, label_name, weight_column_name): argument
374 def loss_fn(logits, target): function
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Dcustom_loss_head.py31 loss_fn, argument
Destimator.py107 def loss_fn(labels, logits, weights=None): function
531 def loss_fn(labels, logits): function
557 def loss_fn(labels, logits): function
/external/tensorflow/tensorflow/contrib/eager/python/examples/l2hmc/
Dmain.py153 loss_fn=l2hmc.compute_loss, argument
Dl2hmc.py345 def loss_and_grads(dynamics, x, loss_fn=compute_loss): argument
/external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/
Dkeras_test.py280 def loss_fn(y_true, y_pred): function
358 def loss_fn(y_true, y_pred): function
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
Destimator.py149 def loss_fn(labels, logits, weights=None): function
/external/tensorflow/tensorflow/contrib/eager/python/
Dsaver_test.py147 loss_fn = lambda: v[0, 0] ** 2 + v[0, 1] ** 2 function
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_eager.py37 def _eager_loss_fn(outputs, targets, loss_fn, output_name): argument
Dtraining_utils.py773 def get_metric_function(metric, output_shape=None, loss_fn=None): argument
/external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_ptb/
Drnn_ptb.py155 def loss_fn(model, inputs, targets, training): function