Searched defs:logits_t (Results 1 – 3 of 3) sorted by relevance
153 void DoCompute(OpKernelContext* ctx, const Tensor& logits_t, in DoCompute()229 const Tensor& logits_t = ctx->input(0); in Compute() local285 const Tensor& logits_t = ctx->input(0); in Compute() local
29 Tensor logits_t(DT_FLOAT, TensorShape({batch_size, num_classes})); in Multinomial() local
1094 def compute_ctc_loss(logits_t, labels_t, label_length_t, logit_length_t, argument