Searched refs:output_logits (Results 1 – 1 of 1) sorted by relevance
219 auto output_logits = output_logits_t->matrix<float>(); in Compute() local223 output_logits.setZero(); in Compute()228 auto do_work = [&resource, &batch_bucketized_features, &output_logits, in Compute()255 output_logits(i, j) = tree_logits[j]; in Compute()