Searched defs:shallow_tree (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/util/ |
| H A D | nest.py | 229 def _yield_flat_up_to(shallow_tree, input_tree): argument
240 def assert_shallow_structure(shallow_tree, input_tree, check_types=True): argument
308 def flatten_up_to(shallow_tree, input_tree): argument
381 def map_structure_up_to(shallow_tree, func, *inputs): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/util/ |
| H A D | nest.py | 1002 def _yield_flat_up_to(shallow_tree, input_tree, is_nested_fn, path=()): argument
1032 def assert_shallow_structure(shallow_tree, argument
1161 def flatten_up_to(shallow_tree, input_tree, check_types=True, argument
1250 def flatten_with_tuple_paths_up_to(shallow_tree, argument
1355 def map_structure_up_to(shallow_tree, func, *inputs, **kwargs): argument
1435 def map_structure_with_tuple_paths_up_to(shallow_tree, func, *inputs, **kwargs): argument
|
| H A D | nest_test.py | 772 def get_paths_and_values(shallow_tree, input_tree): argument
|