Searched refs:input_tree (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/data/util/ |
| D | nest.py | 250 def _yield_flat_up_to(shallow_tree, input_tree): argument
254 _yield_value(input_tree)):
258 yield input_tree
261 def assert_shallow_structure(shallow_tree, input_tree, check_types=True): argument
298 if not is_sequence(input_tree):
301 "Input has type: %s." % type(input_tree))
303 if check_types and not isinstance(input_tree, type(shallow_tree)):
307 % (type(input_tree), type(shallow_tree)))
309 if len(input_tree) != len(shallow_tree):
313 % (len(input_tree), len(shallow_tree)))
[all …]
|
| D | nest_test.py | 306 input_tree = (((2, 2), (3, 3)), ((4, 9), (5, 5)))
308 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree)
313 input_tree = ((("a", 1), (("b", 2), (("c", 3), (("d", 4))))))
316 input_tree)
317 input_tree_flattened = nest.flatten(input_tree)
324 input_tree = ["input_tree"]
326 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree)
328 self.assertEqual(flattened_input_tree, [input_tree])
331 input_tree = ("input_tree_0", "input_tree_1")
333 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree)
[all …]
|
| /external/tensorflow/tensorflow/python/util/ |
| D | nest_test.py | 518 nest.assert_shallow_structure(shallow_tree=inp_abc, input_tree=inp_ab)
550 input_tree = [[[2, 2], [3, 3]], [[4, 9], [5, 5]]]
552 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree)
558 input_tree = [[("a", 1), [("b", 2), [("c", 3), [("d", 4)]]]]]
561 input_tree)
562 input_tree_flattened = nest.flatten(input_tree)
568 input_tree = {"a": 1, "b": {"c": 2}, "d": [3, (4, 5)]}
571 input_tree)
577 input_tree = ab_tuple(a=[0, 1], b=2)
580 input_tree)
[all …]
|
| D | nest.py | 591 def _yield_flat_up_to(shallow_tree, input_tree, path=()): argument
612 yield (path, input_tree)
614 input_tree = dict(_yield_sorted_items(input_tree))
617 input_subtree = input_tree[shallow_key]
624 def assert_shallow_structure(shallow_tree, input_tree, check_types=True): argument
663 if not is_sequence(input_tree):
666 "Input has type: %s." % type(input_tree))
668 if check_types and not isinstance(input_tree, type(shallow_tree)):
672 input_is_namedtuple = _is_namedtuple(input_tree, False)
674 if not _same_namedtuples(shallow_tree, input_tree):
[all …]
|
| /external/tensorflow/tensorflow/tools/compatibility/ |
| D | tf_upgrade_v2_main.py | 113 elif args.input_tree:
121 output_tree = args.input_tree if args.in_place else args.output_tree
123 args.input_tree, output_tree, args.copy_other_files)
|
| D | tf_upgrade.py | 242 elif args.input_tree:
244 args.input_tree, args.output_tree, args.copy_other_files)
|