Home
last modified time | relevance | path

Searched refs:flatten_with_tuple_paths (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/module/
H A Dmodule.py345 for path, child in nest.flatten_with_tuple_paths(structure):
424 leaves = nest.flatten_with_tuple_paths(prop)
/aosp_15_r20/external/tensorflow/tensorflow/python/trackable/
H A Ddata_structures_test.py696 nest.flatten_with_tuple_paths((1, (2,))),
697 nest.flatten_with_tuple_paths(t))
704 nest.flatten_with_tuple_paths(nt),
705 nest.flatten_with_tuple_paths(wrapped_nt))
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dcomposite_tensor_test.py123 result_with_paths = nest.flatten_with_tuple_paths(
H A Dfunc_graph.py124 flattened = nest.flatten_with_tuple_paths(structure)
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Dnest.py1680 def flatten_with_tuple_paths(structure, expand_composites=False): function
H A Dnest_test.py800 nest.flatten_with_tuple_paths(input_tree)]
1154 self.assertEqual(nest.flatten_with_tuple_paths(inputs), expected)