OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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