HomeSort by relevance Sort by last modified time
    Searched refs:nested_splits_lists (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/python/ops/ragged/
ragged_functional_ops.py 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):
89 Appends each `RaggedTensor`'s `nested_splits` to `nested_splits_lists`.
93 nested_splits_lists: An output parameter used to record the `nested_splits`
102 nested_splits_lists.append(value.nested_row_splits
    [all...]
ragged_util.py 78 def assert_splits_match(nested_splits_lists):
86 nested_splits_lists: A list of nested_splits_lists, where each split_list is
96 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)
ragged_dispatch.py 129 nested_splits_lists = [
137 ragged_util.assert_splits_match(nested_splits_lists)):
140 nested_splits_lists[0])

Completed in 111 milliseconds