Home
last modified time | relevance | path

Searched refs:get_structure (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dprefetch_to_device_test.py50 dataset_ops.get_structure(host_dataset),
51 dataset_ops.get_structure(device_dataset)))
75 dataset_ops.get_structure(host_dataset),
76 dataset_ops.get_structure(device_dataset)))
99 dataset_ops.get_structure(host_dataset),
100 dataset_ops.get_structure(device_dataset)))
127 dataset_ops.get_structure(host_dataset),
128 dataset_ops.get_structure(device_dataset)))
175 dataset_ops.get_structure(host_dataset),
176 dataset_ops.get_structure(device_dataset)))
Dcopy_to_device_test.py54 dataset_ops.get_structure(host_dataset),
55 dataset_ops.get_structure(device_dataset)))
79 dataset_ops.get_structure(host_dataset),
80 dataset_ops.get_structure(device_dataset)))
103 dataset_ops.get_structure(host_dataset),
104 dataset_ops.get_structure(device_dataset)))
128 dataset_ops.get_structure(host_dataset),
129 dataset_ops.get_structure(device_dataset)))
153 dataset_ops.get_structure(host_dataset),
154 dataset_ops.get_structure(device_dataset)))
[all …]
Dvariant_test.py35 dataset_ops.get_structure(dataset))
45 dataset_ops.get_structure(dataset))
/external/tensorflow/tensorflow/python/data/experimental/ops/
Ddistribute.py143 dataset_ops.get_structure(input_dataset))
208 for ts in nest.flatten(dataset_ops.get_structure(self._input_dataset))
284 rebatch, dataset_ops.get_structure(input_dataset))
508 for ts in nest.flatten(dataset_ops.get_structure(dataset))
Dwriters.py107 if not dataset_ops.get_structure(dataset).is_compatible_with(
/external/tensorflow/tensorflow/python/data/kernel_tests/
Diterator_test.py422 dataset_ops.get_structure(dataset_3),
423 dataset_ops.get_structure(feedable_iterator)))
480 dataset_ops.get_structure(dataset_3),
481 dataset_ops.get_structure(feedable_iterator)))
800 dataset_ops.get_structure(iterator), expected_element_structure))
834 dataset_ops.get_structure(iterator), expected_element_structure))
881 dataset_ops.get_structure(iterator), expected_element_structure))
Dtest_base.py227 dataset_ops.get_structure(dataset1),
228 dataset_ops.get_structure(dataset2)))
Ddataset_test.py285 dataset_ops.get_structure(dataset), expected_element_structure))
/external/tensorflow/tensorflow/python/data/experimental/
D__init__.py150 from tensorflow.python.data.ops.dataset_ops import get_structure
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.experimental.pbtxt132 name: "get_structure"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.pbtxt160 name: "get_structure"
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py2887 def get_structure(dataset_or_iterator): function
2924 get_structure(dataset_or_iterator))
2941 get_structure(dataset_or_iterator))
2958 get_structure(dataset_or_iterator))
4822 get_structure(input_dataset))
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_v1.py2375 x_shapes = dataset_ops.get_structure(x)
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistribute_strategy_test.py2562 input_shape = dataset_ops.get_structure(dataset)[0].shape[1:]