Searched refs:out_sampled_candidates (Results 1 – 1 of 1) sorted by relevance
60 Tensor* out_sampled_candidates = nullptr; in Compute() local63 &out_sampled_candidates)); in Compute()77 out_sampled_candidates->vec<int64>().data(), num_sampled_); in Compute()