Home
last modified time | relevance | path

Searched refs:get_legacy_output_shapes (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_tensors_test.py50 nest.flatten(dataset_ops.get_legacy_output_shapes(dataset)))
69 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
88 ], [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
103 dataset_ops.get_legacy_output_shapes(dataset))
110 dataset_ops.get_legacy_output_shapes(dataset))
117 dataset_ops.get_legacy_output_shapes(dataset))
124 dataset_ops.get_legacy_output_shapes(dataset))
131 dataset_ops.get_legacy_output_shapes(dataset))
138 dataset_ops.get_legacy_output_shapes(dataset))
148 dataset_ops.get_legacy_output_shapes(dataset))
[all …]
Dfrom_tensor_slices_test.py48 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
72 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
121 ], [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
169 self.assertEqual((), dataset_ops.get_legacy_output_shapes(dataset)["foo"])
170 self.assertEqual((1,), dataset_ops.get_legacy_output_shapes(dataset)["bar"])
Drepeat_test.py42 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
60 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
76 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
Dconcatenate_test.py49 dataset_ops.get_legacy_output_shapes(concatenated),
82 dataset_ops.get_legacy_output_shapes(concatenated))],
Diterator_test.py317 [None], dataset_ops.get_legacy_output_shapes(iterator).as_list())
413 dataset_ops.get_legacy_output_shapes(dataset_3))
472 dataset_ops.get_legacy_output_shapes(dataset_3))
610 dataset_ops.get_legacy_output_shapes(dataset_3))
687 dataset_ops.get_legacy_output_shapes(itr))
726 dataset_ops.get_legacy_output_shapes(dataset_3))
882 dataset_ops.get_legacy_output_shapes(iterator))
Diterator_cluster_test.py57 dataset_ops.get_legacy_output_shapes(dataset_3))
74 dataset_ops.get_legacy_output_shapes(dataset_3))
Dskip_test.py38 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
Dtake_test.py38 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
Dzip_test.py84 dataset_ops.get_legacy_output_shapes(dataset),
Dfilter_test_base.py56 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
/external/tensorflow/tensorflow/contrib/data/python/kernel_tests/
Dassert_element_shape_test.py47 dataset_ops.get_legacy_output_shapes(dataset))
51 dataset_ops.get_legacy_output_shapes(result))
89 dataset_ops.get_legacy_output_shapes(dataset))
95 dataset_ops.get_legacy_output_shapes(result))
121 dataset_ops.get_legacy_output_shapes(dataset))
150 dataset_ops.get_legacy_output_shapes(result))
188 dataset_ops.get_legacy_output_shapes(dataset))
194 dataset_ops.get_legacy_output_shapes(result))
220 dataset_ops.get_legacy_output_shapes(dataset))
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dchoose_fastest_dataset_test.py41 expected_shapes=dataset_ops.get_legacy_output_shapes(dataset))
49 expected_shapes=dataset_ops.get_legacy_output_shapes(dataset))
62 expected_shapes=dataset_ops.get_legacy_output_shapes(dataset_a))
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dscan_test.py112 [None], dataset_ops.get_legacy_output_shapes(dataset)[0][0].as_list())
114 None, dataset_ops.get_legacy_output_shapes(dataset)[0][1].ndims)
116 [], dataset_ops.get_legacy_output_shapes(dataset)[1].as_list())
Dmap_and_batch_test.py89 for shape in dataset_ops.get_legacy_output_shapes(dataset)])
149 [4, 1], dataset_ops.get_legacy_output_shapes(dataset).as_list())
152 [None, 1], dataset_ops.get_legacy_output_shapes(dataset).as_list())
172 [None, 1], dataset_ops.get_legacy_output_shapes(dataset).as_list())
Denumerate_dataset_test.py42 dataset_output_shapes = dataset_ops.get_legacy_output_shapes(dataset)
Dcounter_test.py35 [], dataset_ops.get_legacy_output_shapes(dataset).as_list())
Drestructured_dataset_test.py55 nest.flatten(dataset_ops.get_legacy_output_shapes(new))):
Dmake_tf_record_dataset_test.py160 for shape in nest.flatten(dataset_ops.get_legacy_output_shapes(dataset)):
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dbatching.py82 dataset_output_shapes = dataset_ops.get_legacy_output_shapes(dataset)
121 dataset_output_shapes = dataset_ops.get_legacy_output_shapes(dataset)
365 input_shapes = dataset_ops.get_legacy_output_shapes(input_dataset)
440 dataset_ops.get_legacy_output_shapes(dataset),
527 input_shapes = dataset_ops.get_legacy_output_shapes(dataset)
752 input_shapes = dataset_ops.get_legacy_output_shapes(self._input_dataset)
Dinterleave_ops.py114 output_shapes = dataset_ops.get_legacy_output_shapes(self._data_inputs[0])
119 nest.flatten(dataset_ops.get_legacy_output_shapes(data_input)))
Dwriters.py58 dataset_ops.get_legacy_output_shapes(dataset),
Dparsing_ops.py60 input_dataset_shape = dataset_ops.get_legacy_output_shapes(
/external/tensorflow/tensorflow/contrib/data/python/ops/
Dbatching.py220 dataset_ops.get_legacy_output_shapes(dataset), expected_shapes)
/external/tensorflow/tensorflow/python/tpu/
Ddatasets.py163 dataset_ops.get_legacy_output_shapes(source_dataset))
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py243 output_shapes = nest.map_structure(str, get_legacy_output_shapes(self))
1471 None, get_legacy_output_types(self), get_legacy_output_shapes(self),
1528 get_legacy_output_shapes(dataset), get_legacy_output_classes(dataset))
1897 def get_legacy_output_shapes(dataset_or_iterator): function
2567 input_shapes = get_legacy_output_shapes(self._input_dataset)
2572 nest.flatten(get_legacy_output_shapes(
2922 input_shapes = get_legacy_output_shapes(input_dataset)
3248 nest.flatten(get_legacy_output_shapes(input_dataset)),

12