HomeSort by relevance Sort by last modified time
    Searched refs:along_class (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
softmax_op_functor.h 55 Eigen::DSizes<int, 1> along_class(kClassDim);
59 Eigen::IndexList<Eigen::type2index<kClassDim> > along_class; local
67 auto shifted_logits = (logits - logits.maximum(along_class)
77 .sum(along_class)
89 softmax.device(d) = (softmax * softmax.sum(along_class)
xent_op.h 66 Eigen::array<int, 1> along_class; local
67 along_class[0] = kClassDim;
77 Eigen::IndexList<Eigen::type2index<kClassDim> > along_class; local
87 scratch.reshape(batch_only).device(d) = logits.maximum(along_class);
93 scratch.reshape(batch_only).device(d) = backprop.exp().sum(along_class);
105 .sum(along_class);
sparse_xent_op.h 170 Eigen::array<int, 1> along_class; local
171 along_class[0] = kClassDim;
181 Eigen::IndexList<Eigen::type2index<kClassDim> > along_class; local
191 To32Bit(scratch).device(d) = To32Bit(logits).maximum(along_class);
199 To32Bit(scratch).device(d) = To32Bit(backprop).exp().sum(along_class);
209 To32Bit(backprop).generate(sparse_xent_loss_gen).sum(along_class);

Completed in 67 milliseconds