Searched refs:output_partial_logits (Results 1 – 1 of 1) sorted by relevance
98 auto output_partial_logits = output_partial_logits_t->matrix<float>(); in Compute() local118 output_partial_logits.setZero(); in Compute()122 &cached_node_ids, &output_partial_logits, in Compute()176 output_partial_logits(i, j) = partial_all_logits[j]; in Compute()