Searched refs:batch_indices_ (Results 1 – 1 of 1) sorted by relevance
108 return batch_indices_[i]; in batch_indices()121 std::vector<int64> batch_indices_[N]; variable308 &batch_indices_[i]); in BCastList()390 return batch_indices_[0]; in x_batch_indices()397 return batch_indices_[1]; in y_batch_indices()