Searched refs:one_by_class (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | xent_op.h | 82 Eigen::array<int, 2> one_by_class; in Compute() local
83 one_by_class[0] = 1; in Compute()
84 one_by_class[1] = num_classes; in Compute()
91 Eigen::IndexList<Eigen::type2index<1>, int> one_by_class; in Compute() local
92 one_by_class.set(1, num_classes); in Compute()
101 logits.broadcast(logits_bcast) - scratch.broadcast(one_by_class); in Compute()
114 (scratch.log().eval().broadcast(one_by_class) - backprop)) in Compute()
120 backprop.device(d) = (backprop.exp() / scratch.broadcast(one_by_class)) - in Compute()
|
| D | softmax_op_functor.h | 57 Eigen::DSizes<int, 2> one_by_class(1, num_classes); in Compute()
62 Eigen::IndexList<Eigen::type2index<1>, int> one_by_class; in Compute() local
63 one_by_class.set(1, num_classes); in Compute()
69 .broadcast(one_by_class)); in Compute()
80 .broadcast(one_by_class)); in Compute()
92 .broadcast(one_by_class)); in Compute()
|
| D | sparse_xent_op.h | 199 Eigen::array<int, 2> one_by_class; in Compute() local
200 one_by_class[0] = 1; in Compute()
201 one_by_class[1] = num_classes; in Compute()
208 Eigen::IndexList<Eigen::type2index<1>, int> one_by_class; in Compute() local
209 one_by_class.set(1, num_classes); in Compute()
219 To32Bit(scratch).reshape(batch_by_one).broadcast(one_by_class); in Compute()
|