Home
last modified time | relevance | path

Searched refs:batch_by_one (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsoftmax_op_functor.h56 Eigen::DSizes<int, 2> batch_by_one(batch_size, 1); in Compute()
60 Eigen::IndexList<int, Eigen::type2index<1> > batch_by_one; in Compute() local
61 batch_by_one.set(0, batch_size); in Compute()
68 .reshape(batch_by_one) in Compute()
79 .reshape(batch_by_one) in Compute()
91 .reshape(batch_by_one) in Compute()
Dxent_op.h79 Eigen::array<int, 2> batch_by_one; in Compute() local
80 batch_by_one[0] = batch_size; in Compute()
81 batch_by_one[1] = 1; in Compute()
87 Eigen::IndexList<int, Eigen::type2index<1> > batch_by_one; in Compute() local
88 batch_by_one.set(0, batch_size); in Compute()
Dsparse_xent_op.h196 Eigen::array<int, 2> batch_by_one; in Compute() local
197 batch_by_one[0] = batch_size; in Compute()
198 batch_by_one[1] = 1; in Compute()
204 Eigen::IndexList<int, Eigen::type2index<1> > batch_by_one; in Compute() local
205 batch_by_one.set(0, batch_size); in Compute()
219 To32Bit(scratch).reshape(batch_by_one).broadcast(one_by_class); in Compute()