Searched refs:output_partial_logits (Results 1 – 1 of 1) sorted by relevance
91 auto output_partial_logits = output_partial_logits_t->matrix<float>(); in Compute() local111 output_partial_logits.setZero(); in Compute()115 &cached_node_ids, &output_partial_logits, in Compute()162 output_partial_logits(i, 0) = partial_all_logit; in Compute()