Searched refs:splits_values (Results 1 – 3 of 3) sorted by relevance
267 const auto splits_values = splits.flat<int64>(); in Compute() local277 OP_REQUIRES(context, splits_values(0) == 0, in Compute()279 splits_values(0))); in Compute()280 OP_REQUIRES(context, splits_values(num_batches) == num_values, in Compute()283 splits_values(num_batches), " instead of ", num_values)); in Compute()290 while (idx >= splits_values(batch_idx)) { in Compute()
17 while (idx >= splits_values(batch_idx)) {28 However, if the first element of `splits_values` is not 0, `batch_idx` will
14 while (idx >= splits_values(batch_idx)) {