Searched refs:logits_t (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | multinomial_op.cc | 153 void DoCompute(OpKernelContext* ctx, const Tensor& logits_t, in DoCompute() argument
155 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(logits_t.shape()), in DoCompute()
157 logits_t.shape().DebugString())); in DoCompute()
169 const int64 dim = logits_t.dim_size(i); in DoCompute()
172 "logits.shape = ", logits_t.shape().DebugString(), in DoCompute()
175 const int batch_size = static_cast<int>(logits_t.dim_size(0)); in DoCompute()
176 const int num_classes = static_cast<int>(logits_t.dim_size(1)); in DoCompute()
212 ctx, ctx->eigen_device<Device>(), logits_t.matrix<T>(), in DoCompute()
229 const Tensor& logits_t = ctx->input(0); in Compute() local
231 this->DoCompute(ctx, logits_t, num_samples_t, &generator_); in Compute()
[all …]
|
| D | multinomial_op_test.cc | 29 Tensor logits_t(DT_FLOAT, TensorShape({batch_size, num_classes})); in Multinomial() local
31 logits_t.flat<float>().setRandom(); in Multinomial()
36 .Input(test::graph::Constant(g, logits_t)) in Multinomial()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 768 def compute_ctc_loss(logits_t, labels_t, label_length_t, logit_length_t, argument
771 logits_t.set_shape(logits.shape)
776 logits=logits_t,
|