Searched refs:nested_splits_lists (Results 1 – 3 of 3) sorted by relevance
69 nested_splits_lists = []70 inner_args = _replace_ragged_with_flat_values(args, nested_splits_lists)71 inner_kwargs = _replace_ragged_with_flat_values(kwargs, nested_splits_lists)72 if not nested_splits_lists:76 ragged_util.assert_splits_match(nested_splits_lists)):80 op(*inner_args, **inner_kwargs), nested_splits_lists[0])83 def _replace_ragged_with_flat_values(value, nested_splits_lists): argument102 nested_splits_lists.append(value.nested_row_splits)107 return _replace_ragged_with_flat_values(v, nested_splits_lists)
78 def assert_splits_match(nested_splits_lists): argument96 for splits_list in nested_splits_lists:97 if len(splits_list) != len(nested_splits_lists[0]):101 for splits_list in nested_splits_lists[1:]102 for (s1, s2) in zip(nested_splits_lists[0], splits_list)
129 nested_splits_lists = [137 ragged_util.assert_splits_match(nested_splits_lists)):140 nested_splits_lists[0])