Home
last modified time | relevance | path

Searched refs:variant_tensor (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/python/data/experimental/ops/
Dtesting.py92 variant_tensor = (
97 super(_AssertNextDataset, self).__init__(input_dataset, variant_tensor)
106 variant_tensor = (
110 super(_NonSerializableDataset, self).__init__(input_dataset, variant_tensor)
119 variant_tensor = gen_experimental_dataset_ops.sleep_dataset(
123 super(_SleepDataset, self).__init__(input_dataset, variant_tensor)
Ddistribute.py73 variant_tensor = ged_ops.auto_shard_dataset(
81 super(_AutoShardDataset, self).__init__(input_dataset, variant_tensor)
147 variant_tensor = ged_ops.rebatch_dataset_v2(
152 super(_RebatchDataset, self).__init__(input_dataset, variant_tensor)
286 variant_tensor = ged_ops.rebatch_dataset(
290 super(_LegacyRebatchDataset, self).__init__(input_dataset, variant_tensor)
303 variant_tensor = ged_ops.dataset_from_graph(graph_def)
304 super(_RemoteDataset, self).__init__(variant_tensor)
Derror_ops.py67 variant_tensor = (
73 variant_tensor = (
77 super(_IgnoreErrorsDataset, self).__init__(input_dataset, variant_tensor)
Doptimization.py101 variant_tensor = (
106 super(_ChooseFastestDataset, self).__init__(variant_tensor)
208 variant_tensor = (
219 variant_tensor)
Dsleep.py30 variant_tensor = gen_experimental_dataset_ops.sleep_dataset(
34 super(_SleepDataset, self).__init__(input_dataset, variant_tensor)
Dmatching_files.py34 variant_tensor = ged_ops.matching_files_dataset(self._patterns)
35 super(MatchingFilesDataset, self).__init__(variant_tensor)
Dunique.py62 variant_tensor = gen_experimental_dataset_ops.unique_dataset(
65 super(_UniqueDataset, self).__init__(input_dataset, variant_tensor)
Dsnapshot.py84 variant_tensor = ged_ops.snapshot_dataset(
103 super(_LegacySnapshotDataset, self).__init__(input_dataset, variant_tensor)
239 variant_tensor = ged_ops.snapshot_dataset_v2(
248 super(_SnapshotDataset, self).__init__(input_dataset, variant_tensor)
Dcardinality.py112 variant_tensor = ged_ops.assert_cardinality_dataset(
117 variant_tensor)
Drandom_ops.py38 variant_tensor = gen_experimental_dataset_ops.random_dataset(
40 super(RandomDatasetV2, self).__init__(variant_tensor)
Dstats_ops.py104 variant_tensor = self._op_function(
108 super(_StatsDataset, self).__init__(input_dataset, variant_tensor)
Dprefetching_ops.py204 variant_tensor = gen_dataset_ops.generator_dataset(
212 super(_CopyToDeviceDataset, self).__init__(input_dataset, variant_tensor)
241 variant_tensor = ged_ops.experimental_map_dataset(
247 super(_MapOnGpuDataset, self).__init__(input_dataset, variant_tensor)
Dscan_ops.py128 variant_tensor = gen_experimental_dataset_ops.scan_dataset(
137 variant_tensor = gen_experimental_dataset_ops.scan_dataset(
144 super(_ScanDataset, self).__init__(input_dataset, variant_tensor)
Dthreadpool.py69 variant_tensor = ged_ops.thread_pool_dataset(
73 super(_ThreadPoolDataset, self).__init__(input_dataset, variant_tensor)
Dbatching.py314 variant_tensor = ged_ops.dense_to_sparse_batch_dataset(
320 variant_tensor)
361 variant_tensor = ged_ops.map_and_batch_dataset(
370 super(_MapAndBatchDataset, self).__init__(input_dataset, variant_tensor)
Dshuffle_ops.py43 variant_tensor = gen_dataset_ops.shuffle_and_repeat_dataset(
51 variant_tensor)
Dio.py152 variant_tensor = gen_experimental_dataset_ops.load_dataset(
158 super(_LoadDataset, self).__init__(variant_tensor)
Dgrouping.py282 variant_tensor = ged_ops.experimental_group_by_reducer_dataset(
293 super(_GroupByReducerDataset, self).__init__(input_dataset, variant_tensor)
404 variant_tensor = ged_ops.group_by_window_dataset(
413 super(_GroupByWindowDataset, self).__init__(input_dataset, variant_tensor)
Dparsing_ops.py87 variant_tensor = (
101 super(_ParseExampleDataset, self).__init__(input_dataset, variant_tensor)
Ddata_service_ops.py149 variant_tensor = gen_experimental_dataset_ops.data_service_dataset(
162 variant_tensor = gen_experimental_dataset_ops.data_service_dataset_v2(
176 super(_DataServiceDatasetV2, self).__init__(variant_tensor)
/external/tensorflow/tensorflow/python/data/ops/
Dreaders.py132 variant_tensor = gen_dataset_ops.text_line_dataset(self._filenames,
135 super(_TextLineDataset, self).__init__(variant_tensor)
182 variant_tensor = self._impl._variant_tensor # pylint: disable=protected-access
184 super(TextLineDatasetV2, self).__init__(variant_tensor)
238 variant_tensor = gen_dataset_ops.tf_record_dataset(self._filenames,
241 super(_TFRecordDataset, self).__init__(variant_tensor)
278 variant_tensor = ged_ops.legacy_parallel_interleave_dataset_v2(
289 variant_tensor)
347 variant_tensor = self._impl._variant_tensor # pylint: disable=protected-access
348 super(TFRecordDatasetV2, self).__init__(variant_tensor)
[all …]
Ddataset_ops.py191 def __init__(self, variant_tensor): argument
201 self._variant_tensor_attr = variant_tensor
2312 variant_tensor = self._as_variant_tensor()
2325 super(DatasetV1, self).__init__(variant_tensor)
3165 def __init__(self, input_dataset, variant_tensor): argument
3167 super(UnaryDataset, self).__init__(variant_tensor)
3176 def __init__(self, input_dataset, variant_tensor): argument
3179 input_dataset, variant_tensor)
3195 variant_tensor = gen_dataset_ops.tensor_dataset(
3198 super(TensorDataset, self).__init__(variant_tensor)
[all …]
Doptional_ops.py173 def __init__(self, variant_tensor, element_spec): argument
174 self._variant_tensor = variant_tensor
/external/tensorflow/tensorflow/python/data/util/
Dtraverse_test.py37 variant_tensor = gen_dataset_ops.model_dataset(
39 super(_TestDataset, self).__init__(input_dataset, variant_tensor)
/external/tensorflow/tensorflow/python/data/benchmarks/
Dfrom_tensor_slices_benchmark.py42 variant_tensor = gen_dataset_ops.flat_map_dataset(
48 variant_tensor)

12