Searched refs:in_sampled_candidates (Results 1 – 1 of 1) sorted by relevance
208 const Tensor& in_sampled_candidates = context->input(1); in Compute() local210 TensorShapeUtils::IsVector(in_sampled_candidates.shape()), in Compute()216 for (int64 i = 0; i < in_sampled_candidates.dim_size(0); ++i) { in Compute()217 sampled_candidate_to_pos[in_sampled_candidates.vec<int64>()(i)] = i; in Compute()