Home
last modified time | relevance | path

Searched refs:input_tree (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/data/util/
Dnest.py221 def _yield_flat_up_to(shallow_tree, input_tree): argument
225 _yield_value(input_tree)):
229 yield input_tree
232 def assert_shallow_structure(shallow_tree, input_tree, check_types=True): argument
269 if not is_sequence(input_tree):
272 "Input has type: %s." % type(input_tree))
274 if check_types and not isinstance(input_tree, type(shallow_tree)):
278 % (type(input_tree), type(shallow_tree)))
280 if len(input_tree) != len(shallow_tree):
284 % (len(input_tree), len(shallow_tree)))
[all …]
Dnest_test.py319 input_tree = (((2, 2), (3, 3)), ((4, 9), (5, 5)))
321 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree)
326 input_tree = ((("a", 1), (("b", 2), (("c", 3), (("d", 4))))))
329 input_tree)
330 input_tree_flattened = nest.flatten(input_tree)
337 input_tree = ["input_tree"]
339 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree)
341 self.assertEqual(flattened_input_tree, [input_tree])
344 input_tree = ("input_tree_0", "input_tree_1")
346 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree)
[all …]
/external/tensorflow/tensorflow/python/util/
Dnest_test.py622 input_tree = [[[2, 2], [3, 3]], [[4, 9], [5, 5]]]
624 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree)
630 input_tree = [[("a", 1), [("b", 2), [("c", 3), [("d", 4)]]]]]
633 input_tree)
634 input_tree_flattened = nest.flatten(input_tree)
640 input_tree = {"a": 1, "b": {"c": 2}, "d": [3, (4, 5)]}
643 input_tree)
649 input_tree = ab_tuple(a=[0, 1], b=2)
652 input_tree)
657 input_tree = collections.OrderedDict(
[all …]
Dnest.py952 def _yield_flat_up_to(shallow_tree, input_tree, is_seq, path=()): argument
970 yield (path, input_tree)
972 input_tree = dict(_yield_sorted_items(input_tree))
975 input_subtree = input_tree[shallow_key]
983 input_tree, argument
1027 if not is_seq(input_tree):
1030 "Input has type: %s." % type(input_tree))
1037 if check_types and not isinstance(input_tree, shallow_type):
1041 input_is_namedtuple = _is_namedtuple(input_tree, False)
1043 if not _same_namedtuples(shallow_tree, input_tree):
[all …]
/external/tensorflow/tensorflow/tools/compatibility/
Dtf_upgrade_v2_main.py167 elif args.input_tree:
174 output_tree = args.input_tree if args.in_place else args.output_tree
176 args.input_tree, output_tree, args.copy_other_files)
Dtf_upgrade.py245 elif args.input_tree:
247 args.input_tree, args.output_tree, args.copy_other_files)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.__internal__.nest.pbtxt5 …argspec: "args=[\'shallow_tree\', \'input_tree\', \'check_types\', \'expand_composites\'], varargs…