Searched refs:nested_splits_lists (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_functional_ops.py | 93 nested_splits_lists = []
95 inner_args = _replace_ragged_with_flat_values(args, nested_splits_lists,
97 inner_kwargs = _replace_ragged_with_flat_values(kwargs, nested_splits_lists,
99 if not nested_splits_lists:
111 split_dtypes = set(splits[0].dtype for splits in nested_splits_lists)
118 nested_splits_lists = [
120 for nested_splits in nested_splits_lists]
123 ragged_util.assert_splits_match(nested_splits_lists)):
136 op_output, nested_splits_lists[0], validate=False)
139 def _replace_ragged_with_flat_values(value, nested_splits_lists, argument
[all …]
|
| D | ragged_util.py | 31 def assert_splits_match(nested_splits_lists): argument
49 for splits_list in nested_splits_lists:
50 if len(splits_list) != len(nested_splits_lists[0]):
54 for splits_list in nested_splits_lists[1:]
55 for (s1, s2) in zip(nested_splits_lists[0], splits_list)
|
| D | ragged_dispatch.py | 142 nested_splits_lists = [elt.nested_row_splits for elt in ragged_elts]
148 ragged_util.assert_splits_match(nested_splits_lists)):
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | metrics_utils.py | 541 def _assert_splits_match(nested_splits_lists): argument
559 for splits_list in nested_splits_lists:
560 if len(splits_list) != len(nested_splits_lists[0]):
564 for splits_list in nested_splits_lists[1:]
565 for (s1, s2) in zip(nested_splits_lists[0], splits_list)
|