Searched refs:out_ids (Results 1 – 1 of 1) sorted by relevance
242 Tensor* out_ids = nullptr; in Compute() local245 1, TensorShape({static_cast<int>(ids.size())}), &out_ids)); in Compute()254 out_ids->vec<int64>()(i) = ids[i]; in Compute()