Home
last modified time | relevance | path

Searched refs:_RemoteDataset (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/ops/
Ddistribute.py297 class _RemoteDataset(dataset_ops.DatasetSource): class
304 super(_RemoteDataset, self).__init__(variant_tensor)
348 ds = _RemoteDataset(graph_def, device, dataset.element_spec)
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dauto_shard_dataset_test.py328 ds1 = distribute._RemoteDataset(graph_def, "/device:CPU:0",
330 ds2 = distribute._RemoteDataset(graph_def, "/device:CPU:0",