Searched refs:onehot_labels (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/losses/ |
| D | losses_impl.py | 725 onehot_labels, logits, weights=1.0, label_smoothing=0, scope=None, argument
768 if onehot_labels is None:
773 (logits, onehot_labels, weights)) as scope:
775 onehot_labels = math_ops.cast(onehot_labels, logits.dtype)
776 logits.get_shape().assert_is_compatible_with(onehot_labels.get_shape())
780 array_ops.shape(onehot_labels)[-1], logits.dtype)
783 onehot_labels = onehot_labels * smooth_positives + smooth_negatives
785 onehot_labels = array_ops.stop_gradient(
786 onehot_labels, name="labels_stop_gradient")
788 labels=onehot_labels, logits=logits, name="xentropy")
|
| /external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| D | hvp_test.py | 39 logits=logits, onehot_labels=labels)
51 logits=logits, onehot_labels=labels)
59 logits=logits, onehot_labels=labels)
72 logits=logits, onehot_labels=labels)
|
| D | resnet50_graph_test.py | 113 logits=logits, onehot_labels=labels)
|
| D | resnet50_test.py | 41 logits=logits, onehot_labels=labels)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.losses.pbtxt | 65 …argspec: "args=[\'onehot_labels\', \'logits\', \'weights\', \'label_smoothing\', \'scope\', \'loss…
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients_test.py | 293 logits=logits, onehot_labels=label, reduction=losses.Reduction.NONE)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver_test.py | 2169 onehot_labels = sparse_ops.sparse_to_dense(
2173 labels=onehot_labels, logits=logits, name="xentropy")
|