Home
last modified time | relevance | path

Searched refs:original_dataset (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Ddataset_test.py88 original_dataset = dataset_ops.Dataset.range(5).map(lambda x: x * 2)
89 fn = original_dataset._trace_variant_creation()
93 variant, original_dataset.element_spec)
99 original_dataset = dataset_ops.Dataset.range(5).flat_map(
101 fn = original_dataset._trace_variant_creation()
105 variant, original_dataset.element_spec)
106 self.assertDatasetProduces(revived_dataset, list(original_dataset))
116 original_dataset = readers.TFRecordDataset([file_path],
118 fn = original_dataset._trace_variant_creation()
122 variant, original_dataset.element_spec)
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/
H A Dvalues.py424 original_dataset = dataset._original_dataset
425 serialized = serialize_dataset_to_graph(original_dataset)
429 serialized, original_dataset.element_spec)
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dstandalone.cc167 Dataset::Dataset(DatasetBase* finalized_dataset, DatasetBase* original_dataset, in Dataset() argument
172 original_dataset_(original_dataset), in Dataset()
H A Dstandalone.h115 Dataset(DatasetBase* finalized_dataset, DatasetBase* original_dataset,
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dtraining_generator_v1.py127 original_dataset = None
130 original_dataset = data
323 generator = dataset_ops.make_one_shot_iterator(original_dataset)