Searched refs:_loss_fn (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | step_fn.py | 91 self._loss_fn = loss_fn
99 gradients_fn = backprop.implicit_grad(self._loss_fn)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | head.py | 725 self._loss_fn = loss_fn
748 loss_fn=self._loss_fn,
847 self._loss_fn = loss_fn if loss_fn else _log_loss_with_two_classes
869 loss_fn=self._loss_fn,
1044 self._loss_fn = loss_fn if loss_fn else _softmax_cross_entropy_loss
1114 return self._loss_fn(self._label_ids(labels), logits, weights=weights)
1234 def _loss_fn(labels, logits, weights=None): function
1249 self._loss_fn = _loss_fn
1271 loss_fn=self._loss_fn,
1344 self._loss_fn = loss_fn if loss_fn else _sigmoid_cross_entropy_loss
[all …]
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | target_column.py | 158 self._loss_fn = loss_fn
226 loss_unweighted = self._loss_fn(logits, target)
251 loss_unweighted = self._loss_fn(logits, target)
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/workshop/ |
| D | 2_models.ipynb | 724 "def _loss_fn(logits, labels):\n",
734 " loss = _loss_fn(logits, labels)\n",
|