Searched refs:along_class (Results 1 – 3 of 3) sorted by relevance
75 Eigen::array<int, 1> along_class; in Compute() local76 along_class[0] = kClassDim; in Compute()86 Eigen::IndexList<Eigen::type2index<kClassDim> > along_class; in Compute() local97 logits.broadcast(logits_bcast).maximum(along_class); in Compute()104 scratch.reshape(batch_only).device(d) = backprop.exp().sum(along_class); in Compute()116 .sum(along_class); in Compute()
55 Eigen::DSizes<int, 1> along_class(kClassDim); in Compute()59 Eigen::IndexList<Eigen::type2index<kClassDim> > along_class; in Compute() local66 auto shifted_logits = (logits - logits.maximum(along_class) in Compute()76 .sum(along_class) in Compute()88 softmax.device(d) = (softmax * softmax.sum(along_class) in Compute()
192 Eigen::array<int, 1> along_class; in Compute() local193 along_class[0] = kClassDim; in Compute()203 Eigen::IndexList<Eigen::type2index<kClassDim> > along_class; in Compute() local222 To32Bit(scratch).device(d) = To32Bit(backprop).exp().sum(along_class); in Compute()232 To32Bit(backprop).generate(sparse_xent_loss_gen).sum(along_class); in Compute()