Searched refs:flatten_up_to (Results 1 – 13 of 13) sorted by relevance
/external/tensorflow/tensorflow/python/data/util/ |
D | nest_test.py | 321 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree) 322 flattened_shallow_tree = nest.flatten_up_to(shallow_tree, shallow_tree) 328 input_tree_flattened_as_shallow_tree = nest.flatten_up_to(shallow_tree, 339 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree) 340 flattened_shallow_tree = nest.flatten_up_to(shallow_tree, shallow_tree) 346 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree) 347 flattened_shallow_tree = nest.flatten_up_to(shallow_tree, shallow_tree) 354 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree) 355 flattened_shallow_tree = nest.flatten_up_to(shallow_tree, shallow_tree) 361 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree) [all …]
|
D | nest.py | 300 def flatten_up_to(shallow_tree, input_tree): function 442 flatten_up_to(shallow_tree, input_tree) for input_tree in inputs)
|
D | structure.py | 98 components = nest.flatten_up_to(element_signature, element)
|
/external/tensorflow/tensorflow/python/util/ |
D | nest_test.py | 624 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree) 625 flattened_shallow_tree = nest.flatten_up_to(shallow_tree, shallow_tree) 632 input_tree_flattened_as_shallow_tree = nest.flatten_up_to(shallow_tree, 642 input_tree_flattened_as_shallow_tree = nest.flatten_up_to(shallow_tree, 651 input_tree_flattened_as_shallow_tree = nest.flatten_up_to(shallow_tree, 661 input_tree_flattened_as_shallow_tree = nest.flatten_up_to(shallow_tree, 665 input_tree_flattened_as_shallow_tree = nest.flatten_up_to(shallow_tree, 672 input_tree_flattened_as_shallow_tree = nest.flatten_up_to(shallow_tree, 682 flattened_input_tree = nest.flatten_up_to(shallow_tree, input_tree) 683 flattened_shallow_tree = nest.flatten_up_to(shallow_tree, shallow_tree) [all …]
|
D | nest.py | 1111 def flatten_up_to(shallow_tree, input_tree, check_types=True, function 1465 flatten_up_to( # pylint: disable=g-complex-comprehension
|
/external/tensorflow/tensorflow/tools/api/golden/v2/ |
D | tensorflow.__internal__.nest.pbtxt | 4 name: "flatten_up_to"
|
/external/tensorflow/tensorflow/python/saved_model/ |
D | function_deserialization.py | 67 flatten_inputs = nest.flatten_up_to( 96 flatten_inputs = nest.flatten_up_to(expected_structure, inputs)
|
/external/tensorflow/tensorflow/python/keras/engine/ |
D | compile_utils.py | 346 self._metrics = nest.flatten_up_to(y_pred, self._metrics, check_types=False) 347 self._weighted_metrics = nest.flatten_up_to(
|
D | training_utils_v1.py | 441 batch_outs = nest.flatten_up_to(self._structure, batch_outs) 462 batch_outs = nest.flatten_up_to(self._structure, batch_outs)
|
/external/tensorflow/tensorflow/python/framework/ |
D | composite_tensor_test.py | 165 result = nest.flatten_up_to(s1, s2, expand_composites=expand_composites)
|
/external/tensorflow/tensorflow/python/tpu/ |
D | tpu_feed.py | 863 inputs_part_dims_flat = nest.flatten_up_to(flattened_inputs,
|
/external/tensorflow/tensorflow/python/data/ops/ |
D | dataset_ops.py | 898 flattened_values = nest.flatten_up_to(output_types, values) 4229 flat_padded_shapes = nest.flatten_up_to(input_shapes, padded_shapes)
|
/external/tensorflow/tensorflow/python/eager/ |
D | function.py | 2826 flatten_inputs = nest.flatten_up_to(
|