Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_tensors_test.py52 nest.flatten(dataset_ops.get_legacy_output_shapes(dataset)))
91 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
111 ], [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
203 dataset_ops.get_legacy_output_shapes(dataset)["a"]["aa"])
205 dataset_ops.get_legacy_output_shapes(dataset)["a"]["ab"])
207 dataset_ops.get_legacy_output_shapes(dataset)["b"])
216 self.assertEqual([3], dataset_ops.get_legacy_output_shapes(dataset))
222 self.assertEqual([3], dataset_ops.get_legacy_output_shapes(dataset))
227 self.assertEqual([2, 3], dataset_ops.get_legacy_output_shapes(dataset))
233 self.assertEqual([3], dataset_ops.get_legacy_output_shapes(dataset))
Drepeat_test.py40 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
50 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
67 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
Ddataset_test.py449 dataset_ops.get_legacy_output_shapes(dataset))
456 dataset_ops.get_legacy_output_shapes(dataset))
463 dataset_ops.get_legacy_output_shapes(dataset))
470 dataset_ops.get_legacy_output_shapes(dataset))
477 dataset_ops.get_legacy_output_shapes(dataset))
484 dataset_ops.get_legacy_output_shapes(dataset))
492 dataset_ops.get_legacy_output_shapes(dataset))
498 dataset_output_shapes = dataset_ops.get_legacy_output_shapes(dataset)
517 dataset_ops.get_legacy_output_shapes(dataset))
Dfrom_tensor_slices_test.py52 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
103 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
153 ], [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
199 self.assertEqual((), dataset_ops.get_legacy_output_shapes(dataset)["foo"])
200 self.assertEqual((1,), dataset_ops.get_legacy_output_shapes(dataset)["bar"])
Dconcatenate_test.py50 dataset_ops.get_legacy_output_shapes(concatenated),
84 dataset_ops.get_legacy_output_shapes(concatenated))],
Diterator_test.py320 [None], dataset_ops.get_legacy_output_shapes(iterator).as_list())
417 dataset_ops.get_legacy_output_shapes(dataset_3))
475 dataset_ops.get_legacy_output_shapes(dataset_3))
613 dataset_ops.get_legacy_output_shapes(dataset_3))
690 dataset_ops.get_legacy_output_shapes(itr))
729 dataset_ops.get_legacy_output_shapes(dataset_3))
806 dataset_ops.get_legacy_output_shapes(iterator))
840 dataset_ops.get_legacy_output_shapes(iterator))
887 dataset_ops.get_legacy_output_shapes(iterator))
Dtake_test.py39 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
Dskip_test.py39 [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)])
Denumerate_test.py43 dataset_output_shapes = dataset_ops.get_legacy_output_shapes(dataset)
Diterator_cluster_test.py60 dataset_ops.get_legacy_output_shapes(dataset_3))
77 dataset_ops.get_legacy_output_shapes(dataset_3))
Dzip_test.py86 dataset_ops.get_legacy_output_shapes(dataset),
Dfilter_test.py77 [shape for shape in 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))
50 expected_shapes=dataset_ops.get_legacy_output_shapes(dataset))
64 expected_shapes=dataset_ops.get_legacy_output_shapes(dataset_a))
Dchoose_fastest_branch_dataset_test.py50 expected_shapes=dataset_ops.get_legacy_output_shapes(dataset))
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dinterleave_ops.py126 output_shapes = dataset_ops.get_legacy_output_shapes(self._data_inputs[0])
131 nest.flatten(dataset_ops.get_legacy_output_shapes(data_input)))
Dwriters.py112 dataset_ops.get_legacy_output_shapes(dataset),
Dparsing_ops.py66 input_dataset_shape = dataset_ops.get_legacy_output_shapes(
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dscan_test.py207 [None], dataset_ops.get_legacy_output_shapes(dataset)[0][0].as_list())
209 None, dataset_ops.get_legacy_output_shapes(dataset)[0][1].ndims)
211 [], dataset_ops.get_legacy_output_shapes(dataset)[1].as_list())
Dmap_and_batch_test.py80 for shape in dataset_ops.get_legacy_output_shapes(dataset)])
132 [4, 1], dataset_ops.get_legacy_output_shapes(dataset).as_list())
135 [None, 1], dataset_ops.get_legacy_output_shapes(dataset).as_list())
148 [None, 1], dataset_ops.get_legacy_output_shapes(dataset).as_list())
Dcounter_test.py42 [], dataset_ops.get_legacy_output_shapes(dataset).as_list())
Dmake_tf_record_dataset_test.py170 for shape in nest.flatten(dataset_ops.get_legacy_output_shapes(dataset)):
Drebatch_dataset_test.py156 for ts in nest.flatten(dataset_ops.get_legacy_output_shapes(dataset))
425 dataset_ops.get_legacy_output_shapes(dataset)))
431 nest.flatten(dataset_ops.get_legacy_output_shapes(rebatched_dataset)))
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dtext_vectorization.py409 shape = dataset_ops.get_legacy_output_shapes(data)
415 shape = dataset_ops.get_legacy_output_shapes(data)
418 self.build(dataset_ops.get_legacy_output_shapes(data))
/external/tensorflow/tensorflow/python/tpu/
Ddatasets.py168 dataset_ops.get_legacy_output_shapes(source_dataset))
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dmap_vectorization_benchmark.py186 dataset_ops.get_legacy_output_shapes(base_dataset))

12