Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Ddataset_ops.py225 def __init__(self, variant_tensor): argument
235 self._variant_tensor_attr = variant_tensor
3633 variant_tensor = self._as_variant_tensor()
3646 super(DatasetV1, self).__init__(variant_tensor)
4388 def __init__(self, input_dataset, variant_tensor): argument
4390 super(UnaryDataset, self).__init__(variant_tensor)
4399 def __init__(self, input_dataset, variant_tensor): argument
4402 input_dataset, variant_tensor)
4426 def __init__(self, variant_tensor, element_spec, dataset_shape): argument
4427 self._variant_tensor = variant_tensor
[all …]
H A Dreaders.py155 variant_tensor = gen_dataset_ops.text_line_dataset(
160 super(_TextLineDataset, self).__init__(variant_tensor)
243 variant_tensor = self._impl._variant_tensor # pylint: disable=protected-access
245 super(TextLineDatasetV2, self).__init__(variant_tensor)
307 variant_tensor = gen_dataset_ops.tf_record_dataset(
310 super(_TFRecordDataset, self).__init__(variant_tensor)
358 variant_tensor = ged_ops.legacy_parallel_interleave_dataset_v2(
369 variant_tensor)
471 variant_tensor = self._impl._variant_tensor # pylint: disable=protected-access
472 super(TFRecordDatasetV2, self).__init__(variant_tensor)
[all …]
H A Dload_op.py72 variant_tensor = ged_ops.load_dataset(
78 super(_LoadDataset, self).__init__(variant_tensor)
H A Dmulti_device_iterator_ops.py146 variant_tensor = gen_dataset_ops.generator_dataset(
154 super(_PerDeviceGenerator, self).__init__(variant_tensor)
188 variant_tensor = gen_dataset_ops.generator_dataset(
196 super(_ReincarnatedPerDeviceGenerator, self).__init__(variant_tensor)
H A Doptional_ops.py166 def __init__(self, variant_tensor, element_spec): argument
168 self._variant_tensor = variant_tensor
H A Dsave_op.py80 variant_tensor = ged_ops.save_dataset_v2(
90 super(_SaveDataset, self).__init__(dataset, variant_tensor)
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Dtesting.py130 variant_tensor = (
135 super(_AssertNextDataset, self).__init__(input_dataset, variant_tensor)
167 variant_tensor = (
172 super(_AssertPrevDataset, self).__init__(input_dataset, variant_tensor)
181 variant_tensor = (
185 super(_NonSerializableDataset, self).__init__(input_dataset, variant_tensor)
194 variant_tensor = gen_experimental_dataset_ops.sleep_dataset(
198 super(_SleepDataset, self).__init__(input_dataset, variant_tensor)
H A Ddistribute.py74 variant_tensor = ged_ops.auto_shard_dataset(
82 super(_AutoShardDataset, self).__init__(input_dataset, variant_tensor)
151 variant_tensor = ged_ops.rebatch_dataset_v2(
157 super(_RebatchDataset, self).__init__(input_dataset, variant_tensor)
305 variant_tensor = ged_ops.rebatch_dataset(
310 super(_LegacyRebatchDataset, self).__init__(input_dataset, variant_tensor)
323 variant_tensor = ged_ops.dataset_from_graph(graph_def)
324 super(_RemoteDataset, self).__init__(variant_tensor)
H A Dprefetching_ops.py206 variant_tensor = gen_dataset_ops.generator_dataset(
214 super(_CopyToDeviceDataset, self).__init__(input_dataset, variant_tensor)
244 variant_tensor = ged_ops.experimental_map_dataset(
250 super(_MapOnGpuDataset, self).__init__(input_dataset, variant_tensor)
H A Dbatching.py319 variant_tensor = ged_ops.dense_to_sparse_batch_dataset(
325 variant_tensor)
366 variant_tensor = ged_ops.map_and_batch_dataset(
375 super(_MapAndBatchDataset, self).__init__(input_dataset, variant_tensor)
H A Dsleep.py26 variant_tensor = gen_experimental_dataset_ops.sleep_dataset(
30 super(_SleepDataset, self).__init__(input_dataset, variant_tensor)
H A Derror_ops.py61 variant_tensor = (
66 super(_IgnoreErrorsDataset, self).__init__(input_dataset, variant_tensor)
H A Dmatching_files.py30 variant_tensor = ged_ops.matching_files_dataset(self._patterns)
31 super(MatchingFilesDataset, self).__init__(variant_tensor)
H A Dcardinality.py108 variant_tensor = ged_ops.assert_cardinality_dataset(
113 variant_tensor)
H A Dshuffle_ops.py39 variant_tensor = gen_dataset_ops.shuffle_and_repeat_dataset(
47 variant_tensor)
H A Dfrom_list.py63 variant_tensor = gen_experimental_dataset_ops.list_dataset(
68 super(_ListDataset, self).__init__(variant_tensor)
H A Dsnapshot.py77 variant_tensor = ged_ops.snapshot_dataset(
96 super(_LegacySnapshotDataset, self).__init__(input_dataset, variant_tensor)
H A Dparsing_ops.py84 variant_tensor = (
98 super(_ParseExampleDataset, self).__init__(input_dataset, variant_tensor)
H A Dreaders.py805 variant_tensor = gen_experimental_dataset_ops.csv_dataset_v2(
817 super(CsvDatasetV2, self).__init__(variant_tensor)
1192 variant_tensor = gen_experimental_dataset_ops.sql_dataset(
1195 super(SqlDatasetV2, self).__init__(variant_tensor)
H A Dgrouping.py270 variant_tensor = ged_ops.experimental_group_by_reducer_dataset(
281 super(_GroupByReducerDataset, self).__init__(input_dataset, variant_tensor)
/aosp_15_r20/external/federated-compute/fcp/artifact_building/
H A Dgraph_helpers_test.py249 variant_tensor = tf.data.experimental.to_variant(tf.data.Dataset.range(3))
256 [variant_tensor],
260 self.assertIs(input_map['foo'], variant_tensor)
264 variant_tensor = tf.data.experimental.to_variant(tf.data.Dataset.range(3))
272 [variant_tensor, variant_tensor],
/aosp_15_r20/external/federated-compute/fcp/tensorflow/
H A Dexternal_dataset.py45 variant_tensor = gen_external_dataset_py.ExternalDataset(
47 super(ExternalDataset, self).__init__(variant_tensor)
/aosp_15_r20/external/tensorflow/tensorflow/python/data/benchmarks/
H A Dfrom_tensor_slices_benchmark.py39 variant_tensor = gen_dataset_ops.flat_map_dataset(
45 variant_tensor)
/aosp_15_r20/external/tensorflow/tensorflow/python/data/util/
H A Dtraverse_test.py38 variant_tensor = gen_dataset_ops.model_dataset(
40 super(_TestDataset, self).__init__(input_dataset, variant_tensor)
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/
H A Dvalues.py396 variant_tensor = ged_ops.dataset_from_graph(graph_def)
397 super(_RemoteDataset, self).__init__(variant_tensor)

12