Home
last modified time | relevance | path

Searched refs:_GeneratorDataset (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_generator_test.py430 dataset = dataset_ops._GeneratorDataset(
Dmulti_device_iterator_test.py423 dataset = dataset_ops._GeneratorDataset(
Diterator_test.py950 dataset = dataset_ops._GeneratorDataset(1, init_fn, next_fn, finalize_fn,
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py1012 return _GeneratorDataset(dummy_arg, get_iterator_id_fn, generator_next_fn,
3628 class _GeneratorDataset(DatasetSource): class
3679 super(_GeneratorDataset, self).__init__(variant_tensor)