Searched refs:probs_batch_indices (Results 1 – 1 of 1) sorted by relevance
191 const auto& probs_batch_indices = bcast.y_batch_indices(); in operator ()() local206 prob = probs(probs_batch_indices[batch_idx]); in operator ()()