Searched refs:along_row (Results 1 – 1 of 1) sorted by relevance
142 Eigen::array<int, 1> along_row; in Compute() local143 along_row[0] = 1; in Compute()145 Eigen::IndexList<Eigen::type2index<1> > along_row; in Compute()148 To32Bit(maximum).device(d) = To32Bit(logits).maximum(along_row); in Compute()