Searched refs:logits_in (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | xent_op.cc | 44 const Tensor& logits_in = context->input(0); in Compute() local
47 TensorShape shape_in = logits_in.shape(); in Compute()
49 BCast bcast(BCast::FromShape(logits_in.shape()), in Compute()
51 if (!logits_in.IsSameSize(labels_in)) { in Compute()
55 logits_in.shape().DebugString(), in Compute()
82 if (logits_in.IsSameSize(labels_in)) { in Compute()
85 Eigen::array<Eigen::DenseIndex, 2>{1, 1}, logits_in.matrix<T>(), in Compute()
92 logits_in.template shaped<T, 2>(bcast.x_reshape()), in Compute()
|
| D | softmax_op.cc | 63 const Tensor& logits_in = context->input(0); in Compute() local
64 OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(logits_in.shape()), in Compute()
66 logits_in.shape().DebugString())); in Compute()
69 {0}, 0, logits_in.shape(), &softmax_out)); in Compute()
70 if (logits_in.NumElements() > 0) { in Compute()
72 functor(context->eigen_device<Device>(), logits_in.flat_inner_dims<T>(), in Compute()
|
| D | softmax_op_gpu.cu.cc | 140 auto logits_in = logits_in_.flat_inner_dims<T>(); in Compute() local
141 const int rows = logits_in.dimension(0); in Compute()
142 const int cols = logits_in.dimension(1); in Compute()
|