Home | History | Annotate | Download | only in kernels

Lines Matching defs:label

48 // logits, and the label indices.  For each minibatch entry, ignoring
51 // loss[j] = (log(sum_exp_logits) - logits[j]) * 1{ j == label }
72 const Index label = tensorflow::internal::SubtleMustCopy(labels_(batch));
73 if (!FastBoundsCheck(label, max_depth_)) {
76 return TF_PREDICT_FALSE(label == depth)
89 // This generator takes the exponentiated logits, their sums, and the label
92 // exp_logits[j] / sum_exp_logits - 1{ j == label }
112 const Index label = tensorflow::internal::SubtleMustCopy(labels_(batch));
113 if (!FastBoundsCheck(label, max_depth_)) {
116 T subtract = TF_PREDICT_FALSE(depth == label) ? T(1.0) : T(0.0);