Searched refs:flatten_up_to (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/python/data/util/ |
D | nest_test.py | 308 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree) 309 flattened_shallow_tree = nest.flatten_up_to(shallow_tree, shallow_tree) 315 input_tree_flattened_as_shallow_tree = nest.flatten_up_to(shallow_tree, 326 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree) 327 flattened_shallow_tree = nest.flatten_up_to(shallow_tree, shallow_tree) 333 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree) 334 flattened_shallow_tree = nest.flatten_up_to(shallow_tree, shallow_tree) 341 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree) 342 flattened_shallow_tree = nest.flatten_up_to(shallow_tree, shallow_tree) 348 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree) [all …]
|
D | nest.py | 329 def flatten_up_to(shallow_tree, input_tree): function 470 all_flattened_up_to = [flatten_up_to(shallow_tree, input_tree)
|
D | structure.py | 344 flat_value = nest.flatten_up_to(self._nested_structure, value) 360 flat_value = nest.flatten_up_to(self._nested_structure, value)
|
/external/tensorflow/tensorflow/python/util/ |
D | nest_test.py | 552 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree) 553 flattened_shallow_tree = nest.flatten_up_to(shallow_tree, shallow_tree) 560 input_tree_flattened_as_shallow_tree = nest.flatten_up_to(shallow_tree, 570 input_tree_flattened_as_shallow_tree = nest.flatten_up_to(shallow_tree, 579 input_tree_flattened_as_shallow_tree = nest.flatten_up_to(shallow_tree, 589 input_tree_flattened_as_shallow_tree = nest.flatten_up_to(shallow_tree, 593 input_tree_flattened_as_shallow_tree = nest.flatten_up_to(shallow_tree, 600 input_tree_flattened_as_shallow_tree = nest.flatten_up_to(shallow_tree, 610 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree) 611 flattened_shallow_tree = nest.flatten_up_to(shallow_tree, shallow_tree) [all …]
|
D | nest.py | 702 def flatten_up_to(shallow_tree, input_tree, check_types=True): function 1028 flat_value_lists = [flatten_up_to(shallow_tree, input_tree, check_types)
|
/external/tensorflow/tensorflow/python/saved_model/ |
D | function_deserialization.py | 62 flatten_inputs = nest.flatten_up_to(expected_structure, inputs) 89 flatten_inputs = nest.flatten_up_to(expected_structure, inputs)
|
/external/tensorflow/tensorflow/contrib/data/python/ops/ |
D | batching.py | 202 flat_new_shapes = nest.flatten_up_to(original_shapes, expected_shapes)
|
/external/tensorflow/tensorflow/python/tpu/ |
D | tpu_feed.py | 838 inputs_part_dims_flat = nest.flatten_up_to(flattened_inputs,
|
/external/tensorflow/tensorflow/python/data/experimental/ops/ |
D | batching.py | 537 flat_new_shapes = nest.flatten_up_to(output_types, output_shapes)
|
/external/tensorflow/tensorflow/python/data/ops/ |
D | dataset_ops.py | 446 flattened_values = nest.flatten_up_to(output_types, values) 2923 flat_padded_shapes = nest.flatten_up_to(input_shapes, padded_shapes)
|
/external/tensorflow/tensorflow/python/eager/ |
D | function.py | 1139 flatten_inputs = nest.flatten_up_to(
|