Searched refs:lengths_to_splits (Results 1 – 4 of 4) sorted by relevance
65 def lengths_to_splits(lengths): function
136 splits.append(ragged_util.lengths_to_splits(masked_row_lengths))157 masked_splits = ragged_util.lengths_to_splits(masked_lengths)374 result_splits.append(ragged_util.lengths_to_splits(output_lengths))
235 stack_splits = ragged_util.lengths_to_splits(stack_lengths)
408 ragged_util.lengths_to_splits(dim_size), splits)