Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/data/util/
H A Dnest.py240 def assert_shallow_structure(shallow_tree, input_tree, check_types=True): function
304 assert_shallow_structure(shallow_branch, input_branch,
377 assert_shallow_structure(shallow_tree, input_tree)
446 assert_shallow_structure(shallow_tree, input_tree)
H A Dnest_test.py301 nest.assert_shallow_structure(inp_abc, inp_ab)
310 nest.assert_shallow_structure(inp_ab2, inp_ab1)
311 nest.assert_shallow_structure(inp_ab2, inp_ab1, check_types=False)
320 nest.assert_shallow_structure(inp_ab2, inp_ab1)
324 nest.assert_shallow_structure(inp_ab, inp_ba)
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Dnest.py1032 def assert_shallow_structure(shallow_tree, function
1155 assert_shallow_structure(shallow_branch, input_branch,
1240 assert_shallow_structure(shallow_tree,
1347 assert_shallow_structure(shallow_tree,
1511 assert_shallow_structure(
1590 assert_shallow_structure(to_traverse, structure,
H A Dnest_test.py586 nest.assert_shallow_structure(inp_abc, inp_ab)
595 nest.assert_shallow_structure(inp_ab2, inp_ab1)
596 nest.assert_shallow_structure(inp_ab2, inp_ab1, check_types=False)
603 nest.assert_shallow_structure(inp_ab2, inp_ab1)
607 nest.assert_shallow_structure(inp_ab, inp_ba)
613 nest.assert_shallow_structure(inp_shallow, inp_deep, check_types=False)
614 nest.assert_shallow_structure(inp_shallow, inp_deep, check_types=True)
620 nest.assert_shallow_structure(inp_shallow, inp_deep, check_types=False)
621 nest.assert_shallow_structure(inp_shallow, inp_deep, check_types=True)
769 nest.assert_shallow_structure(shallow_tree, input_tree)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dcomposite_tensor_test.py203 nest.assert_shallow_structure(s1, s2, expand_composites=expand_composites)
224 nest.assert_shallow_structure(s1, s2, expand_composites=False)
239 nest.assert_shallow_structure(s1, s2, expand_composites=expand_composites)
255 nest.assert_shallow_structure(
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_feed.py115 nest.assert_shallow_structure(dequeues, dims)
/aosp_15_r20/external/tensorflow/tensorflow/python/trackable/
H A Ddata_structures_test.py726 nest.assert_shallow_structure((None, None), t)